rename job.py to async_job.py (#378)

Author: jean-lucas

nucleus/__init__.py

@@ -65,6 +65,7 @@
     Segment,
     SegmentationAnnotation,
 )
+from .async_job import AsyncJob
 from .camera_params import CameraParams
 from .connection import Connection
 from .constants import (
@@ -115,7 +116,6 @@
     NotFoundError,
     NucleusAPIError,
 )
-from .job import AsyncJob
 from .logger import logger
 from .model import Model
 from .model_run import ModelRun

nucleus/job.py renamed to nucleus/async_job.py

@@ -8,12 +8,12 @@
 
 import requests
 
+from nucleus.async_job import AsyncJob
 from nucleus.constants import (
     JOB_CREATION_TIME_KEY,
     JOB_LAST_KNOWN_STATUS_KEY,
     JOB_TYPE_KEY,
 )
-from nucleus.job import AsyncJob
 
 
 def entropy(name, model_run, client):

nucleus/autocurate.py

@@ -4,7 +4,7 @@
 import requests
 
 from nucleus.annotation_uploader import AnnotationUploader, PredictionUploader
-from nucleus.job import AsyncJob
+from nucleus.async_job import AsyncJob
 from nucleus.prediction import Prediction, from_json
 from nucleus.track import Track
 from nucleus.url_utils import sanitize_string_args

nucleus/dataset.py

@@ -1,9 +1,9 @@
 from enum import Enum
 from typing import TYPE_CHECKING, Dict, Optional
 
+from .async_job import AsyncJob
 from .camera_params import CameraParams
 from .constants import CAMERA_PARAMS_KEY
-from .job import AsyncJob
 
 if TYPE_CHECKING:
     from . import NucleusClient

nucleus/metadata_manager.py

@@ -2,9 +2,9 @@
 
 import requests
 
+from .async_job import AsyncJob
 from .constants import METADATA_KEY, MODEL_TAGS_KEY, NAME_KEY, REFERENCE_ID_KEY
 from .dataset import Dataset
-from .job import AsyncJob
 from .model_run import ModelRun
 from .prediction import (
     BoxPrediction,

nucleus/model.py

@@ -19,7 +19,7 @@
 
 from nucleus.annotation import check_all_mask_paths_remote
 from nucleus.annotation_uploader import PredictionUploader
-from nucleus.job import AsyncJob
+from nucleus.async_job import AsyncJob
 from nucleus.utils import (
     format_prediction_response,
     serialize_and_write_to_presigned_url,

nucleus/model_run.py

@@ -7,10 +7,10 @@
 import requests
 
 from nucleus.annotation import Annotation
+from nucleus.async_job import AsyncJob
 from nucleus.constants import EXPORT_FOR_TRAINING_KEY, EXPORTED_ROWS, ITEMS_KEY
 from nucleus.dataset_item import DatasetItem
 from nucleus.errors import NucleusAPIError
-from nucleus.job import AsyncJob
 from nucleus.prediction import from_json as prediction_from_json
 from nucleus.scene import Scene
 from nucleus.utils import (

nucleus/slice.py

@@ -1,7 +1,7 @@
 from typing import List
 
+from nucleus.async_job import AsyncJob
 from nucleus.connection import Connection
-from nucleus.job import AsyncJob
 
 from .constants import EVAL_FUNCTION_KEY, SCENARIO_TEST_ID_KEY, EntityLevel
 from .data_transfer_objects.eval_function import (

nucleus/validate/client.py

@@ -6,8 +6,8 @@
 
 import nucleus
 from nucleus import annotation
+from nucleus.async_job import JobError
 from nucleus.dataset import Dataset
-from nucleus.job import JobError
 
 FLAGS = flags.FLAGS
 

scripts/load_test.py

@@ -15,9 +15,9 @@
     SegmentationAnnotation,
 )
 from nucleus.annotation import SceneCategoryAnnotation
+from nucleus.async_job import AsyncJob, JobError
 from nucleus.constants import ERROR_PAYLOAD
 from nucleus.errors import DuplicateIDError
-from nucleus.job import AsyncJob, JobError
 from nucleus.scene import VideoScene
 
 from .helpers import (

tests/test_annotation.py

@@ -3,8 +3,8 @@
 import pytest
 
 from nucleus import DatasetItem, autocurate
+from nucleus.async_job import AsyncJob
 from nucleus.constants import ERROR_PAYLOAD
-from nucleus.job import AsyncJob
 from nucleus.prediction import BoxPrediction
 from tests.helpers import (
     TEST_BOX_PREDICTIONS,

tests/test_autocurate.py

@@ -11,6 +11,7 @@
     PolygonAnnotation,
     SegmentationAnnotation,
 )
+from nucleus.async_job import AsyncJob, JobError
 from nucleus.constants import (
     ANNOTATIONS_KEY,
     BOX_TYPE,
@@ -27,7 +28,6 @@
     UPDATED_ITEMS,
 )
 from nucleus.errors import NucleusAPIError
-from nucleus.job import AsyncJob, JobError
 from nucleus.scene import LidarScene, VideoScene
 
 from .helpers import (

tests/test_dataset.py

@@ -1,14 +1,14 @@
 import pytest
 
 from nucleus import DatasetItem
+from nucleus.async_job import AsyncJob
 from nucleus.constants import (
     BACKFILL_JOB_KEY,
     ERROR_PAYLOAD,
     JOB_ID_KEY,
     MESSAGE_KEY,
     STATUS_KEY,
 )
-from nucleus.job import AsyncJob
 
 from .helpers import (
     TEST_DATASET_NAME,

tests/test_indexing.py

@@ -15,9 +15,9 @@
     Segment,
     SegmentationPrediction,
 )
+from nucleus.async_job import AsyncJob, JobError
 from nucleus.constants import ERROR_PAYLOAD
 from nucleus.errors import DuplicateIDError
-from nucleus.job import AsyncJob, JobError
 from nucleus.scene import VideoScene
 
 from .helpers import (

tests/test_prediction.py

@@ -11,6 +11,7 @@
     LidarScene,
     VideoScene,
 )
+from nucleus.async_job import JobError
 from nucleus.constants import (
     ANNOTATIONS_KEY,
     FRAME_RATE_KEY,
@@ -30,7 +31,6 @@
     URL_KEY,
     VIDEO_URL_KEY,
 )
-from nucleus.job import JobError
 from nucleus.scene import flatten
 
 from .helpers import (

tests/test_scene.py

@@ -4,13 +4,13 @@
 import requests
 
 from nucleus import BoxAnnotation, BoxPrediction, Dataset, NucleusClient, Slice
+from nucleus.async_job import AsyncJob
 from nucleus.constants import (
     ANNOTATIONS_KEY,
     BOX_TYPE,
     ITEM_KEY,
     PREDICTIONS_KEY,
 )
-from nucleus.job import AsyncJob
 
 from .helpers import (
     TEST_BOX_ANNOTATIONS,

tests/test_slice.py

@@ -1,6 +1,6 @@
 import pytest
 
-from nucleus.job import AsyncJob
+from nucleus.async_job import AsyncJob
 from nucleus.validate.scenario_test_evaluation import (
     ScenarioTestEvaluation,
     ScenarioTestItemEvaluation,