Fix linting issues

Author: whoseoyster

openlayer/__init__.py

@@ -26,12 +26,12 @@
 import tarfile
 import tempfile
 import time
+import urllib.parse
 import uuid
 import warnings
 from typing import Dict, Optional, Tuple
 
 import pandas as pd
-import urllib.parse
 import yaml
 
 from . import api, constants, exceptions, utils
@@ -922,7 +922,7 @@ def create_inference_pipeline(
  {"task_type": task_type.value, **reference_dataset_config}
  )
 
- with tempfile.TemporaryDirectory() as tmp_dir: 
+ with tempfile.TemporaryDirectory() as tmp_dir:
  # Copy relevant files to tmp dir if reference dataset is provided
  if reference_dataset_config_file_path is not None:
  utils.write_yaml(
@@ -1135,9 +1135,7 @@ def publish_batch_data(
  with tarfile.open(tar_file_path, mode="w:gz") as tar:
  tar.add(tmp_dir, arcname=os.path.basename("batch_data"))
 
- payload = {
- "performGroundTruthMerge": False,
- }
+ payload = {"performGroundTruthMerge": False}
 
  presigned_url_query_params_dict = {
  "earliestTimestamp": int(earliest_timestamp),
@@ -1157,8 +1155,10 @@ def publish_batch_data(
  body=payload,
  storage_uri_key="storageUri",
  method="POST",
- presigned_url_endpoint=f"inference-pipelines/{inference_pipeline_id}/presigned-url",
- presigned_url_query_params=presigned_url_query_params
+ presigned_url_endpoint=(
+ f"inference-pipelines/{inference_pipeline_id}/presigned-url"
+ ),
+ presigned_url_query_params=presigned_url_query_params,
  )
 
  print("Batch of data published!")
@@ -1232,6 +1232,6 @@ def publish_ground_truths(
  storage_uri_key="storageUri",
  method="POST",
  presigned_url_endpoint=f"inference-pipelines/{inference_pipeline_id}/presigned-url",
- presigned_url_query_params=presigned_url_query_params
+ presigned_url_query_params=presigned_url_query_params,
  )
  print("Ground truths published!")

openlayer/api.py

@@ -28,7 +28,6 @@
 from requests_toolbelt import MultipartEncoder, MultipartEncoderMonitor
 from tqdm import tqdm
 from tqdm.utils import CallbackIOWrapper
-import urllib.parse
 
 from . import constants
 from .exceptions import ExceptionMap, OpenlayerException