get_job from job_id (#206)

* add get_job

* add inttest

* bump ver and update changelog

* release is blocked on deploy

* retry ci

* fix inttest breaking ci

Author: drakejwong

CHANGELOG.md

@@ -4,22 +4,41 @@ All notable changes to the [Nucleus Python Client](https://github.com/scaleapi/n
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
-## [0.5.1](https://github.com/scaleapi/nucleus-python-client/releases/tag/v0.5.1) - 2021-01-11
+## [0.5.4](https://github.com/scaleapi/nucleus-python-client/releases/tag/v0.5.4) - 2022-01-28
+
+### Added
+- Add `NucleusClient.get_job` to retrieve `AsyncJob`s by job ID
+
+## [0.5.3](https://github.com/scaleapi/nucleus-python-client/releases/tag/v0.5.3) - 2022-01-25
+
+### Added
+- Add average precision to polygon metrics
+- Add mean average precision to polygon metrics
+
+## [0.5.2](https://github.com/scaleapi/nucleus-python-client/releases/tag/v0.5.2) - 2022-01-20
+
+### Added
+- Add `Dataset.delete_scene`
+
+### Fixed
+- Removed `Shapely` dependency
+
+## [0.5.1](https://github.com/scaleapi/nucleus-python-client/releases/tag/v0.5.1) - 2022-01-11
 
 ### Fixed
 - Updated dependencies for full Python 3.6 compatibility
 
-## [0.5.0](https://github.com/scaleapi/nucleus-python-client/releases/tag/v0.5.0) - 2021-01-10
+## [0.5.0](https://github.com/scaleapi/nucleus-python-client/releases/tag/v0.5.0) - 2022-01-10
 
 ### Added
 - `nucleus.metrics` module for computing metrics between Nucleus `Annotation` and `Prediction` objects.
 
-## [0.4.5](https://github.com/scaleapi/nucleus-python-client/releases/tag/v0.4.5) - 2021-01-07
+## [0.4.5](https://github.com/scaleapi/nucleus-python-client/releases/tag/v0.4.5) - 2022-01-07
 
 ### Added
 - `Dataset.scenes` property that fetches the Scale-generated ID, reference ID, type, and metadata of all scenes in the Dataset.
 
-## [0.4.4](https://github.com/scaleapi/nucleus-python-client/releases/tag/v0.4.4) - 2021-01-04
+## [0.4.4](https://github.com/scaleapi/nucleus-python-client/releases/tag/v0.4.4) - 2022-01-04
 
 ### Added
 - `Slice.export_raw_items()` method that fetches accessible (signed) URLs for all items in the Slice.

nucleus/__init__.py

@@ -275,6 +275,22 @@ def get_dataset(self, dataset_id: str) -> Dataset:
         """
         return Dataset(dataset_id, self)
 
+    def get_job(self, job_id: str) -> AsyncJob:
+        """Fetches a dataset by its ID.
+
+        Parameters:
+            job_id: The ID of the dataset to fetch.
+
+        Returns:
+            :class:`AsyncJob`: The Nucleus async job as an object.
+        """
+        payload = self.make_request(
+            payload={},
+            route=f"job/{job_id}/info",
+            requests_command=requests.get,
+        )
+        return AsyncJob.from_json(payload=payload, client=self)
+
     def get_model(self, model_id: str) -> Model:
         """Fetches a model by its ID.
 

pyproject.toml

@@ -21,7 +21,7 @@ exclude = '''
 
 [tool.poetry]
 name = "scale-nucleus"
-version = "0.5.3"
+version = "0.5.4"
 description = "The official Python client library for Nucleus, the Data Platform for AI"
 license =  "MIT"
 authors = ["Scale AI Nucleus Team <nucleusapi@scaleapi.com>"]

tests/test_jobs.py

@@ -23,8 +23,15 @@ def test_repr(test_object: any):
     )
 
 
-def test_job_creation_and_listing(CLIENT):
+def test_job_creation_and_listing_and_retrieval(CLIENT):
     jobs = CLIENT.list_jobs()
 
+    if not jobs:
+        return
+
     for job in jobs:
         assert eval(str(job)) == job
+
+    fetch_id = jobs[0].job_id
+    fetched_job = CLIENT.get_job(fetch_id)
+    assert fetched_job.status() == jobs[0].status()