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

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

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

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

@@ -1,6 +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):
@@ -10,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.
@@ -97,17 +118,16 @@ class ModelInfo:
 
         return identifier
 
-    def add_file_path(self, file_path: str):
+    def add_file(self, file: ModelFile):
         """
         add a path to list of known path.
         detect depending on file type further actions.
-        :param file_path: file_path
+        :param file: ModelFile
         :return: None
         """
-        path, extension = os.path.splitext(file_path)
-        if path == "trained_model":
-            self.file_format = extension.lstrip('.')
-        self.files.append(file_path)
+        if file.name == "trained_model":
+            self.file_format = file.extension
+        self.files.append(file)
 
     def to_json(self):
         """

wizata_dsapi-1.3.29/wizata_dsapi/version.py

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

{wizata_dsapi-1.3.28 → 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.28
+Version: 1.3.29
 Summary: Wizata Data Science Toolkit
 Author: Wizata S.A.
 Author-email: info@wizata.com

wizata_dsapi-1.3.28/wizata_dsapi/version.py

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