Lint fixes

Author: dandye

iocs/v1alpha/batch_get_iocs.py

@@ -15,7 +15,7 @@
 # limitations under the License.
 """Get multiple IoCs from Chronicle."""
 
-from typing import List
+from typing import List, Mapping, Any
 
 from common import regions
 from google.auth.transport import requests
@@ -30,7 +30,7 @@ def batch_get_iocs(
     proj_region: str,
     ioc_values: List[str],
     ioc_type: str,
-) -> dict:
+) -> Mapping[str, Any]:
   """Get multiple IoCs by their values from Chronicle.
 
     Args:

iocs/v1alpha/get_ioc.py

@@ -28,7 +28,7 @@ def get_ioc(
     proj_region: str,
     ioc_value: str,
     ioc_type: str,
-  ) -> Mapping[str, Any]:
+) -> Mapping[str, Any]:
   """Get a single IoC by its value from Chronicle.
 
   Args:
@@ -59,8 +59,10 @@ def get_ioc(
   """
   base_url_with_region = regions.url_always_prepend_region(
       CHRONICLE_API_BASE_URL, proj_region)
+  # pylint: disable=line-too-long
   instance = f"projects/{proj_id}/locations/{proj_region}/instances/{proj_instance}"
   url = f"{base_url_with_region}/v1alpha/{instance}/iocs/{ioc_type}/{ioc_value}"
+  # pylint: enable=line-too-long
 
   response = http_session.request("GET", url)
   if response.status_code >= 400:

iocs/v1alpha/get_ioc_state.py

@@ -17,6 +17,7 @@
 
 from common import regions
 from google.auth.transport import requests
+from typing import Any, Mapping
 
 CHRONICLE_API_BASE_URL = "https://chronicle.googleapis.com"
 
@@ -28,7 +29,7 @@ def get_ioc_state(
     proj_region: str,
     ioc_value: str,
     ioc_type: str,
-) -> dict:
+) -> Mapping[str, Any]:
   """Get the state of an IoC by its value from Chronicle.
 
     Args: