wizata-dsapi 1.3.27__tar.gz → 1.3.29__tar.gz

{wizata_dsapi-1.3.27 → wizata_dsapi-1.3.29}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizata_dsapi
-Version: 1.3.27
+Version: 1.3.29
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

{wizata_dsapi-1.3.27 → wizata_dsapi-1.3.29}/wizata_dsapi/mlmodel.py

@@ -1,5 +1,7 @@
 from typing import List, Iterator, Union
+import os
 from .api_dto import ApiDto
+from datetime import datetime
 
 
 def get_bool(obj, name: str):
@@ -9,6 +11,26 @@ def get_bool(obj, name: str):
         return bool(obj[name])
 
 
+class ModelFile:
+    """
+    define a model file.
+    """
+
+    def __init__(self,
+                 full_path: str = None,
+                 path: str = None,
+                 name: str = None,
+                 extension: str = None,
+                 last_modified: datetime = None,
+                 size: int = None):
+        self.full_path = full_path
+        self.path = path
+        self.name = name
+        self.extension = extension
+        self.last_modified = last_modified
+        self.size = size
+
+
 class ModelInfo:
     """
     define a pointer to a machine learning model.
@@ -47,6 +69,33 @@ class ModelInfo:
         self.has_target_feat = False
         self.label_counts = 0
 
+    @classmethod
+    def split_identifier(cls, identifier: str):
+        """
+        split the identifier into four parts
+        :param identifier: identifier.
+        :return: key, twin, property, alias
+        """
+
+        # extract alias
+        if "@" in identifier:
+            split_identifier_for_alias = identifier.split("@")
+            identifier = split_identifier_for_alias[0]
+            alias = split_identifier_for_alias[1]
+        else:
+            alias = None
+
+        # extract key, twin, property
+        identifiers = identifier.split('.')
+        key = identifiers[0]
+        twin_hardware_id = None
+        if len(identifiers) > 1:
+            twin_hardware_id = identifiers[1]
+        property_value = None
+        if len(identifiers) > 2:
+            property_value = identifiers[2]
+
+        return key, twin_hardware_id, property_value, alias
 
     def identifier(self, include_alias: bool = False) -> str:
         """
@@ -69,6 +118,17 @@ class ModelInfo:
 
         return identifier
 
+    def add_file(self, file: ModelFile):
+        """
+        add a path to list of known path.
+        detect depending on file type further actions.
+        :param file: ModelFile
+        :return: None
+        """
+        if file.name == "trained_model":
+            self.file_format = file.extension
+        self.files.append(file)
+
     def to_json(self):
         """
         convert this entity in a dict that can be json serializable

wizata_dsapi-1.3.29/wizata_dsapi/version.py

@@ -0,0 +1 @@
+__version__ = "1.3.29"

{wizata_dsapi-1.3.27 → wizata_dsapi-1.3.29}/wizata_dsapi.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wizata_dsapi
-Version: 1.3.27
+Version: 1.3.29
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

wizata_dsapi-1.3.27/wizata_dsapi/version.py

@@ -1 +0,0 @@
-__version__ = "1.3.27"