atlasai-dstoolkit-client 0.0.10__py3-none-any.whl → 0.0.11__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (6) hide show
  1. atlasai/toolkit/model.py +9 -3
  2. {atlasai_dstoolkit_client-0.0.10.dist-info → atlasai_dstoolkit_client-0.0.11.dist-info}/METADATA +1 -1
  3. {atlasai_dstoolkit_client-0.0.10.dist-info → atlasai_dstoolkit_client-0.0.11.dist-info}/RECORD +6 -6
  4. {atlasai_dstoolkit_client-0.0.10.dist-info → atlasai_dstoolkit_client-0.0.11.dist-info}/LICENSE.txt +0 -0
  5. {atlasai_dstoolkit_client-0.0.10.dist-info → atlasai_dstoolkit_client-0.0.11.dist-info}/WHEEL +0 -0
  6. {atlasai_dstoolkit_client-0.0.10.dist-info → atlasai_dstoolkit_client-0.0.11.dist-info}/top_level.txt +0 -0
atlasai/toolkit/model.py CHANGED
@@ -32,6 +32,12 @@ class ModelPredictWrapper:
32
32
  def __init__(self, model):
33
33
  self._model = model
34
34
 
35
+ def __repr__(self):
36
+ return f'Model {self._model.name} - version: {self._model.version}'
37
+
38
+ def __str__(self):
39
+ return f'Model {self._model.name} - version: {self._model.version}'
40
+
35
41
  def predict(
36
42
  self,
37
43
  data: Union[dict, str, pd.DataFrame] = None,
@@ -100,15 +106,15 @@ class ModelSearch:
100
106
  self.aliases = aliases if isinstance(aliases, list) else [aliases]
101
107
 
102
108
  def search(self):
103
- search = {}
109
+ _search = {}
104
110
  filters = ['name', 'version', 'source', 'tags', 'aliases']
105
111
  for f in filters:
106
112
  if not getattr(self, f, None):
107
113
  continue
108
- search[f] = getattr(self, f)
114
+ _search[f] = getattr(self, f)
109
115
 
110
116
  _get_access_token()
111
- results = client.get_models(search=search, tabular=False)
117
+ results = client.get_models(search=_search, tabular=False)
112
118
  return ModelResults([ModelPredictWrapper(r) for r in results])
113
119
 
114
120
  class ModelGet:
{atlasai_dstoolkit_client-0.0.10.dist-info → atlasai_dstoolkit_client-0.0.11.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: atlasai-dstoolkit-client
3
- Version: 0.0.10
3
+ Version: 0.0.11
4
4
  Summary: UNKNOWN
5
5
  Home-page: UNKNOWN
6
6
  Author: AtlasAI SWE
{atlasai_dstoolkit_client-0.0.10.dist-info → atlasai_dstoolkit_client-0.0.11.dist-info}/RECORD RENAMED
@@ -6,11 +6,11 @@ atlasai/toolkit/fabric.py,sha256=6aFR2PGQc9P3Qa07WdBg9eKoUzU8n2y_-gGjYcyMrWY,192
6
6
  atlasai/toolkit/feature.py,sha256=VKwX_yAwwMo1WuLZ_RbKVJEH4L1PAoYEPzWa0lH9ats,2828
7
7
  atlasai/toolkit/init.py,sha256=JkdJ6QGdYWrq65jgz2pn5RYXUeUe2Ez88_-eMf5CNi0,1100
8
8
  atlasai/toolkit/login.py,sha256=n4ydfo9qCsmbZq6er1xeljBD76vdTJGjbhYHMmOyDbQ,3061
9
- atlasai/toolkit/model.py,sha256=6gmIVx7joCpVxSQbXlmjkxAjreL_J104laAgKKgGOo0,3525
9
+ atlasai/toolkit/model.py,sha256=SBbYZ84Cdw-zut9WogSBpC7WuuXFgtDTVFXXod9FVQY,3729
10
10
  atlasai/toolkit/requests.py,sha256=X86nIo07hAjUlilZcZ1lV8RB7KOsTKbTGtcY_SpFEXY,1223
11
11
  atlasai/toolkit/utils.py,sha256=lYh3P2XOshRgHCjFeXJ0FOJWQW64sddgx8c2kL6Wqwc,1566
12
- atlasai_dstoolkit_client-0.0.10.dist-info/LICENSE.txt,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
13
- atlasai_dstoolkit_client-0.0.10.dist-info/METADATA,sha256=h9oUexHohAl4WiRMqQX2UiXytdzdYmBeGPJ5BhM4thg,1405
14
- atlasai_dstoolkit_client-0.0.10.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
15
- atlasai_dstoolkit_client-0.0.10.dist-info/top_level.txt,sha256=HRTbErU8nmHFDaJJ5R_XYbwpt21dqdjDpSva8xyy_0k,8
16
- atlasai_dstoolkit_client-0.0.10.dist-info/RECORD,,
12
+ atlasai_dstoolkit_client-0.0.11.dist-info/LICENSE.txt,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
13
+ atlasai_dstoolkit_client-0.0.11.dist-info/METADATA,sha256=QRfqzqe1Rk3J0ntjLMw-P5jk5PuvG6pAFkyBkvvVw-c,1405
14
+ atlasai_dstoolkit_client-0.0.11.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
15
+ atlasai_dstoolkit_client-0.0.11.dist-info/top_level.txt,sha256=HRTbErU8nmHFDaJJ5R_XYbwpt21dqdjDpSva8xyy_0k,8
16
+ atlasai_dstoolkit_client-0.0.11.dist-info/RECORD,,