dtlpy 1.115.44__py3-none-any.whl → 1.117.6__py3-none-any.whl

dtlpy/entities/feature.py

@@ -1,124 +1,124 @@
-import attr
-import traceback
-from collections import namedtuple
-from enum import Enum
-
-from .. import repositories, entities
-from ..services.api_client import ApiClient
-
-
-@attr.s
-class Feature(entities.BaseEntity):
-    """
-    Com entity
-    """
-    # platform
-    id = attr.ib()
-    entity_id = attr.ib()
-    url = attr.ib(repr=False)
-    created_at = attr.ib()
-    feature_set_id = attr.ib()
-    version = attr.ib()
-    value = attr.ib()
-    parent_id = attr.ib()
-    project_id = attr.ib()
-    org_id = attr.ib()
-    creator = attr.ib()
-
-    # sdk
-    _client_api = attr.ib(type=ApiClient, repr=False)
-    _repositories = attr.ib(repr=False)
-
-    ################
-    # repositories #
-    ################
-    @_repositories.default
-    def set_repositories(self):
-        reps = namedtuple('repositories',
-                          field_names=['features', 'features_sets'])
-        features_repo = repositories.Features(client_api=self._client_api)
-        r = reps(features=features_repo,
-                 features_sets=repositories.FeatureSets(client_api=self._client_api))
-        return r
-
-    @property
-    def features(self):
-        assert isinstance(self._repositories.features, repositories.Features)
-        return self._repositories.features
-
-    @staticmethod
-    def _protected_from_json(_json, client_api, is_fetched=True):
-        """
-        Same as from_json but with try-except to catch if error
-        :param _json:
-        :param client_api:
-        :return:
-        """
-        try:
-            feature = Feature.from_json(_json=_json,
-                                        client_api=client_api,
-                                        is_fetched=is_fetched)
-            status = True
-        except Exception:
-            feature = traceback.format_exc()
-            status = False
-        return status, feature
-
-    @classmethod
-    def from_json(cls, _json, client_api, is_fetched=True):
-        """
-        Build a Feature entity object from a json
-
-        :param is_fetched: is Entity fetched from Platform
-        :param _json: _json response from host
-        :param client_api: client_api
-        :return: Feature object
-        """
-        inst = cls(
-            id=_json.get('id', None),
-            feature_set_id=_json.get('featureSetId', None),
-            entity_id=_json.get('entityId', None),
-            url=_json.get('url', None),
-            project_id=_json.get('project', None),
-            created_at=_json.get('createdAt', None),
-            version=_json.get('version', None),
-            value=_json.get('value', None),
-            parent_id=_json.get('parentId', None),
-            client_api=client_api,
-            org_id=_json.get('org', None),
-            creator=_json.get('creator', None),
-        )
-        inst.is_fetched = is_fetched
-        return inst
-
-    def to_json(self):
-        """
-        Returns platform _json format of object
-
-        :return: platform json format of object
-        :rtype: dict
-        """
-
-        _json = {'createdAt': self.created_at,
-                 'entityId': self.entity_id,
-                 'id': self.id,
-                 'featureSetId': self.feature_set_id,
-                 'url': self.url,
-                 'project': self.project_id,
-                 'creator': self.creator,
-                 'version': self.version,
-                 'value': self.value,
-                 }
-        if self.parent_id is not None:
-            _json['parentId'] = self.parent_id
-        if self.org_id is not None:
-            _json['org'] = self.org_id
-        return _json
-
-    def delete(self):
-        """
-        Delete Feature Vector object
-
-        :return: True
-        """
-        return self.features.delete(feature_id=self.id)
+import attr
+import traceback
+from collections import namedtuple
+from enum import Enum
+
+from .. import repositories, entities
+from ..services.api_client import ApiClient
+
+
+@attr.s
+class Feature(entities.BaseEntity):
+    """
+    Com entity
+    """
+    # platform
+    id = attr.ib()
+    entity_id = attr.ib()
+    url = attr.ib(repr=False)
+    created_at = attr.ib()
+    feature_set_id = attr.ib()
+    version = attr.ib()
+    value = attr.ib()
+    parent_id = attr.ib()
+    project_id = attr.ib()
+    org_id = attr.ib()
+    creator = attr.ib()
+
+    # sdk
+    _client_api = attr.ib(type=ApiClient, repr=False)
+    _repositories = attr.ib(repr=False)
+
+    ################
+    # repositories #
+    ################
+    @_repositories.default
+    def set_repositories(self):
+        reps = namedtuple('repositories',
+                          field_names=['features', 'features_sets'])
+        features_repo = repositories.Features(client_api=self._client_api)
+        r = reps(features=features_repo,
+                 features_sets=repositories.FeatureSets(client_api=self._client_api))
+        return r
+
+    @property
+    def features(self):
+        assert isinstance(self._repositories.features, repositories.Features)
+        return self._repositories.features
+
+    @staticmethod
+    def _protected_from_json(_json, client_api, is_fetched=True):
+        """
+        Same as from_json but with try-except to catch if error
+        :param _json:
+        :param client_api:
+        :return:
+        """
+        try:
+            feature = Feature.from_json(_json=_json,
+                                        client_api=client_api,
+                                        is_fetched=is_fetched)
+            status = True
+        except Exception:
+            feature = traceback.format_exc()
+            status = False
+        return status, feature
+
+    @classmethod
+    def from_json(cls, _json, client_api, is_fetched=True):
+        """
+        Build a Feature entity object from a json
+
+        :param is_fetched: is Entity fetched from Platform
+        :param _json: _json response from host
+        :param client_api: client_api
+        :return: Feature object
+        """
+        inst = cls(
+            id=_json.get('id', None),
+            feature_set_id=_json.get('featureSetId', None),
+            entity_id=_json.get('entityId', None),
+            url=_json.get('url', None),
+            project_id=_json.get('project', None),
+            created_at=_json.get('createdAt', None),
+            version=_json.get('version', None),
+            value=_json.get('value', None),
+            parent_id=_json.get('parentId', None),
+            client_api=client_api,
+            org_id=_json.get('org', None),
+            creator=_json.get('creator', None),
+        )
+        inst.is_fetched = is_fetched
+        return inst
+
+    def to_json(self):
+        """
+        Returns platform _json format of object
+
+        :return: platform json format of object
+        :rtype: dict
+        """
+
+        _json = {'createdAt': self.created_at,
+                 'entityId': self.entity_id,
+                 'id': self.id,
+                 'featureSetId': self.feature_set_id,
+                 'url': self.url,
+                 'project': self.project_id,
+                 'creator': self.creator,
+                 'version': self.version,
+                 'value': self.value,
+                 }
+        if self.parent_id is not None:
+            _json['parentId'] = self.parent_id
+        if self.org_id is not None:
+            _json['org'] = self.org_id
+        return _json
+
+    def delete(self):
+        """
+        Delete Feature Vector object
+
+        :return: True
+        """
+        return self.features.delete(feature_id=self.id)

dtlpy/entities/feature_set.py

@@ -1,145 +1,152 @@
-from enum import Enum
-
-import attr
-import traceback
-from collections import namedtuple
-
-from .. import repositories, entities
-from ..services.api_client import ApiClient
-
-
-class FeatureEntityType(str, Enum):
-    """Available types for Feature Set entities"""
-    ITEM = 'item'
-    ANNOTATION = 'annotation'
-    DATASET = 'dataset'
-
-
-@attr.s
-class FeatureSet(entities.BaseEntity):
-    """
-    Com entity
-    """
-    # platform
-    id = attr.ib()
-    name = attr.ib()
-    url = attr.ib(repr=False)
-    creator = attr.ib(repr=False)
-    created_at = attr.ib()
-    updated_by = attr.ib()
-    size = attr.ib()
-    set_type = attr.ib()
-    entity_type = attr.ib()
-    project_id = attr.ib()
-    model_id = attr.ib()
-    org_id = attr.ib()
-
-    # sdk
-    _client_api = attr.ib(type=ApiClient, repr=False)
-    _repositories = attr.ib(repr=False)
-
-    ################
-    # repositories #
-    ################
-    @_repositories.default
-    def set_repositories(self):
-        reps = namedtuple('repositories',
-                          field_names=['feature_sets', 'features'])
-        feature_sets_repo = repositories.FeatureSets(client_api=self._client_api,
-                                                     project_id=self.project_id)
-        features_repo = repositories.Features(client_api=self._client_api,
-                                              project_id=self.project_id,
-                                              feature_set=self, )
-        r = reps(feature_sets=feature_sets_repo,
-                 features=features_repo)
-        return r
-
-    @property
-    def feature_sets(self):
-        assert isinstance(self._repositories.feature_sets, repositories.FeatureSets)
-        return self._repositories.feature_sets
-
-    @property
-    def features(self):
-        assert isinstance(self._repositories.features, repositories.Features)
-        return self._repositories.features
-
-    @staticmethod
-    def _protected_from_json(_json, client_api, is_fetched=True):
-        """
-        Same as from_json but with try-except to catch if error
-
-        :param _json: entity's object json
-        :param client_api:
-        :return:
-        """
-        try:
-            feature_set = FeatureSet.from_json(_json=_json,
-                                               client_api=client_api,
-                                               is_fetched=is_fetched)
-            status = True
-        except Exception:
-            feature_set = traceback.format_exc()
-            status = False
-        return status, feature_set
-
-    @classmethod
-    def from_json(cls, _json, client_api, is_fetched=True):
-        """
-        Build a Feature Set entity object from a json
-
-        :param is_fetched: is Entity fetched from Platform
-        :param _json: entity's object json
-        :param client_api: client_api
-        :return: Feature object
-        """
-        inst = cls(
-            id=_json.get('id', None),
-            name=_json.get('name', None),
-            set_type=_json.get('type', None),
-            entity_type=_json.get('entityType', None),
-            size=_json.get('size', None),
-            url=_json.get('url', None),
-            project_id=_json.get('project', None),
-            model_id=_json.get('modelId', None),
-            created_at=_json.get('createdAt', None),
-            creator=_json.get('creator', None),
-            updated_by=_json.get('updatedBy', None),
-            client_api=client_api,
-            org_id=_json.get('org', None),
-        )
-        inst.is_fetched = is_fetched
-        return inst
-
-    def to_json(self):
-        """
-        Returns platform _json format of object
-
-        :return: platform json format of object
-        :rtype: dict
-        """
-
-        _json = {'id': self.id,
-                 'type': self.set_type,
-                 'entityType': self.entity_type,
-                 'project': self.project_id,
-                 'modelId': self.model_id,
-                 'creator': self.creator,
-                 'createdAt': self.created_at,
-                 'updatedBy': self.updated_by,
-                 'name': self.name,
-                 'size': self.size,
-                 'url': self.url}
-        if self.org_id is not None:
-            _json['org'] = self.org_id
-
-        return _json
-
-    def delete(self):
-        """
-        Delete the feature set
-
-        :return: success
-        :rtype: bool
-        """
-        return self.feature_sets.delete(feature_set_id=self.id)
+from enum import Enum
+
+import attr
+import traceback
+from collections import namedtuple
+
+from .. import repositories, entities
+from ..services.api_client import ApiClient
+
+
+class FeatureEntityType(str, Enum):
+    """Available types for Feature Set entities"""
+    ITEM = 'item'
+    ANNOTATION = 'annotation'
+    DATASET = 'dataset'
+
+
+@attr.s
+class FeatureSet(entities.BaseEntity):
+    """
+    Com entity
+    """
+    # platform
+    id = attr.ib()
+    name = attr.ib()
+    url = attr.ib(repr=False)
+    creator = attr.ib(repr=False)
+    created_at = attr.ib()
+    updated_by = attr.ib()
+    size = attr.ib()
+    set_type = attr.ib()
+    entity_type = attr.ib()
+    project_id = attr.ib()
+    model_id = attr.ib()
+    org_id = attr.ib()
+
+    # sdk
+    _client_api = attr.ib(type=ApiClient, repr=False)
+    _repositories = attr.ib(repr=False)
+
+    ################
+    # repositories #
+    ################
+    @_repositories.default
+    def set_repositories(self):
+        reps = namedtuple('repositories',
+                          field_names=['feature_sets', 'features'])
+        feature_sets_repo = repositories.FeatureSets(client_api=self._client_api,
+                                                     project_id=self.project_id)
+        features_repo = repositories.Features(client_api=self._client_api,
+                                              project_id=self.project_id,
+                                              feature_set=self, )
+        r = reps(feature_sets=feature_sets_repo,
+                 features=features_repo)
+        return r
+
+    @property
+    def feature_sets(self):
+        assert isinstance(self._repositories.feature_sets, repositories.FeatureSets)
+        return self._repositories.feature_sets
+
+    @property
+    def features(self):
+        assert isinstance(self._repositories.features, repositories.Features)
+        return self._repositories.features
+
+    @staticmethod
+    def _protected_from_json(_json, client_api, is_fetched=True):
+        """
+        Same as from_json but with try-except to catch if error
+
+        :param _json: entity's object json
+        :param client_api:
+        :return:
+        """
+        try:
+            feature_set = FeatureSet.from_json(_json=_json,
+                                               client_api=client_api,
+                                               is_fetched=is_fetched)
+            status = True
+        except Exception:
+            feature_set = traceback.format_exc()
+            status = False
+        return status, feature_set
+
+    @classmethod
+    def from_json(cls, _json, client_api, is_fetched=True):
+        """
+        Build a Feature Set entity object from a json
+
+        :param is_fetched: is Entity fetched from Platform
+        :param _json: entity's object json
+        :param client_api: client_api
+        :return: Feature object
+        """
+        inst = cls(
+            id=_json.get('id', None),
+            name=_json.get('name', None),
+            set_type=_json.get('type', None),
+            entity_type=_json.get('entityType', None),
+            size=_json.get('size', None),
+            url=_json.get('url', None),
+            project_id=_json.get('project', None),
+            model_id=_json.get('modelId', None),
+            created_at=_json.get('createdAt', None),
+            creator=_json.get('creator', None),
+            updated_by=_json.get('updatedBy', None),
+            client_api=client_api,
+            org_id=_json.get('org', None),
+        )
+        inst.is_fetched = is_fetched
+        return inst
+
+    def to_json(self):
+        """
+        Returns platform _json format of object
+
+        :return: platform json format of object
+        :rtype: dict
+        """
+
+        _json = {'id': self.id,
+                 'type': self.set_type,
+                 'entityType': self.entity_type,
+                 'project': self.project_id,
+                 'modelId': self.model_id,
+                 'creator': self.creator,
+                 'createdAt': self.created_at,
+                 'updatedBy': self.updated_by,
+                 'name': self.name,
+                 'size': self.size,
+                 'url': self.url}
+        if self.org_id is not None:
+            _json['org'] = self.org_id
+
+        return _json
+
+    def update(self):
+        """
+        Update feature set
+
+        :return: entities.FeatureSet
+        """
+        return self.feature_sets.update(self)
+    def delete(self):
+        """
+        Delete the feature set
+
+        :return: success
+        :rtype: bool
+        """
+        return self.feature_sets.delete(feature_set_id=self.id)