esd-services-api-client 2.2.1__tar.gz → 2.2.3__tar.gz

{esd_services_api_client-2.2.1 → esd_services_api_client-2.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: esd-services-api-client
-Version: 2.2.1
+Version: 2.2.3
 Summary: Python clients for ESD services
 Home-page: https://github.com/SneaksAndData/esd-services-api-client
 License: Apache 2.0

esd_services_api_client-2.2.3/esd_services_api_client/_version.py

@@ -0,0 +1 @@
+__version__ = '2.2.3'

{esd_services_api_client-2.2.1 → esd_services_api_client-2.2.3}/esd_services_api_client/crystal/_connector.py

@@ -37,6 +37,7 @@ from esd_services_api_client.crystal._models import (
     AlgorithmRequest,
     AlgorithmConfiguration,
     RequestLifeCycleStage,
+    ParentRequest,
 )
 
 T = TypeVar("T")  # pylint: disable=C0103
@@ -124,6 +125,7 @@ class CrystalConnector:
             RequestLifeCycleStage.FAILED,
             RequestLifeCycleStage.SCHEDULING_TIMEOUT,
             RequestLifeCycleStage.DEADLINE_EXCEEDED,
+            RequestLifeCycleStage.CANCELLED,
         ]
 
     @classmethod
@@ -154,6 +156,7 @@ class CrystalConnector:
         algorithm: str,
         payload: Dict,
         custom_config: Optional[AlgorithmConfiguration] = None,
+        parent_request: Optional[ParentRequest] = None,
         tag: Optional[str] = None,
     ) -> str:
         """
@@ -162,6 +165,7 @@ class CrystalConnector:
         :param algorithm: Name of a connected algorithm.
         :param payload: Algorithm payload.
         :param custom_config: Customized config for this run.
+        :param parent_request: Parent request for this run.
         :param tag: Client-side submission identifier.
         :return: Request identifier assigned to the job by Crystal.
         """
@@ -176,6 +180,7 @@ class CrystalConnector:
             algorithm_name=algorithm,
             algorithm_parameters=payload,
             custom_configuration=custom_config,
+            parent_request=parent_request,
             tag=tag,
         ).to_dict()
 

{esd_services_api_client-2.2.1 → esd_services_api_client-2.2.3}/esd_services_api_client/crystal/_models.py

@@ -36,6 +36,7 @@ class RequestLifeCycleStage(Enum):
     SCHEDULING_TIMEOUT = "SCHEDULING_TIMEOUT"
     DEADLINE_EXCEEDED = "DEADLINE_EXCEEDED"
     THROTTLED = "THROTTLED"
+    CANCELLED = "CANCELLED"
 
 
 @dataclass_json(letter_case=LetterCase.CAMEL)
@@ -133,6 +134,17 @@ class AlgorithmConfiguration(DataClassJsonMixin):
     speculative_attempts: Optional[int] = None
 
 
+@dataclass_json(letter_case=LetterCase.CAMEL)
+@dataclass
+class ParentRequest(DataClassJsonMixin):
+    """
+    Used to specify crystal parent job for a new crystal job.
+    """
+
+    request_id: Optional[str] = None
+    algorithm_name: Optional[str] = None
+
+
 @dataclass_json(letter_case=LetterCase.CAMEL)
 @dataclass
 class AlgorithmRequest(DataClassJsonMixin):
@@ -143,4 +155,5 @@ class AlgorithmRequest(DataClassJsonMixin):
     algorithm_parameters: Dict
     algorithm_name: Optional[str] = None
     custom_configuration: Optional[AlgorithmConfiguration] = None
+    parent_request: Optional[ParentRequest] = None
     tag: Optional[str] = None

{esd_services_api_client-2.2.1 → esd_services_api_client-2.2.3}/esd_services_api_client/nexus/telemetry/recorder.py

@@ -59,18 +59,26 @@ class TelemetryRecorder(NexusCoreObject):
                 entity_name=entity_name,
                 run_id=run_id,
             )
-            self._storage_client.save_data_as_blob(
-                data=entity_to_record,
-                blob_path=DataSocket(
-                    alias="telemetry",
-                    data_path=f"{self._telemetry_base_path}/{entity_name}/{run_id}",
-                    data_format="null",
-                ).parse_data_path(),
-                serialization_format=DictJsonSerializationFormat
-                if isinstance(entity_to_record, dict)
-                else DataFrameParquetSerializationFormat,
-                overwrite=True,
-            )
+            if not isinstance(entity_to_record, dict) and not isinstance(
+                entity_to_record, pd.DataFrame
+            ):
+                self._logger.warning(
+                    "Unsupported data type: {telemetry_entity_type}. Telemetry recording skipped.",
+                    telemetry_entity_type=type(entity_to_record),
+                )
+            else:
+                self._storage_client.save_data_as_blob(
+                    data=entity_to_record,
+                    blob_path=DataSocket(
+                        alias="telemetry",
+                        data_path=f"{self._telemetry_base_path}/{entity_name}/{run_id}",
+                        data_format="null",
+                    ).parse_data_path(),
+                    serialization_format=DictJsonSerializationFormat
+                    if isinstance(entity_to_record, dict)
+                    else DataFrameParquetSerializationFormat,
+                    overwrite=True,
+                )
 
         telemetry_tasks = [
             asyncio.create_task(

{esd_services_api_client-2.2.1 → esd_services_api_client-2.2.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "esd-services-api-client"
-version = "2.2.1"
+version = "2.2.3"
 description = "Python clients for ESD services"
 authors = ["ECCO Sneaks & Data <esdsupport@ecco.com>"]
 maintainers = ['GZU <gzu@ecco.com>', 'JRB <ext-jrb@ecco.com>', 'VISA <visa@ecco.com>']

esd_services_api_client-2.2.1/esd_services_api_client/_version.py

@@ -1 +0,0 @@
-__version__ = '2.2.1'