seekrai 0.4.1__py3-none-any.whl → 0.4.2__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 (8) hide show
  1. seekrai/filemanager.py +4 -3
  2. seekrai/types/deployments.py +1 -0
  3. seekrai/types/finetune.py +1 -0
  4. {seekrai-0.4.1.dist-info → seekrai-0.4.2.dist-info}/METADATA +1 -1
  5. {seekrai-0.4.1.dist-info → seekrai-0.4.2.dist-info}/RECORD +8 -8
  6. {seekrai-0.4.1.dist-info → seekrai-0.4.2.dist-info}/LICENSE +0 -0
  7. {seekrai-0.4.1.dist-info → seekrai-0.4.2.dist-info}/WHEEL +0 -0
  8. {seekrai-0.4.1.dist-info → seekrai-0.4.2.dist-info}/entry_points.txt +0 -0
seekrai/filemanager.py CHANGED
@@ -192,7 +192,7 @@ class DownloadManager:
192
192
  method="GET",
193
193
  url=url,
194
194
  ),
195
- stream=True,
195
+ stream=False,
196
196
  )
197
197
 
198
198
  try:
@@ -221,9 +221,10 @@ class DownloadManager:
221
221
  num_bytes_downloaded = response.num_bytes_downloaded
222
222
 
223
223
  # Raise exception if remote file size does not match downloaded file size
224
- if os.stat(temp_file.name).st_size != file_size:
224
+ temp_file_size = os.stat(temp_file.name).st_size
225
+ if temp_file_size != file_size:
225
226
  DownloadError(
226
- f"Downloaded file size `{pbar.n}` bytes does not match "
227
+ f"Downloaded file size `{temp_file_size}` bytes does not match "
227
228
  f"remote file size `{file_size}` bytes."
228
229
  )
229
230
 
seekrai/types/deployments.py CHANGED
@@ -31,6 +31,7 @@ class DeploymentProcessor(str, enum.Enum):
31
31
  GAUDI2 = "GAUDI2"
32
32
  GAUDI3 = "GAUDI3"
33
33
  A100 = "A100"
34
+ A10 = "A10"
34
35
  H100 = "H100"
35
36
  XEON = "XEON"
36
37
  NVIDIA = "NVIDIA" # TODO - this doesnt make sense with A100, etc.
seekrai/types/finetune.py CHANGED
@@ -111,6 +111,7 @@ class AcceleratorType(str, Enum):
111
111
  GAUDI2 = "GAUDI2"
112
112
  GAUDI3 = "GAUDI3"
113
113
  A100 = "A100"
114
+ A10 = "A10"
114
115
  H100 = "H100"
115
116
 
116
117
 
{seekrai-0.4.1.dist-info → seekrai-0.4.2.dist-info}/METADATA RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.1
2
2
  Name: seekrai
3
- Version: 0.4.1
3
+ Version: 0.4.2
4
4
  Summary: Python client for SeekrAI
5
5
  Home-page: https://gitlab.cb.ntent.com/ml/seekr-py
6
6
  License: Apache-2.0
{seekrai-0.4.1.dist-info → seekrai-0.4.2.dist-info}/RECORD RENAMED
@@ -4,7 +4,7 @@ seekrai/abstract/api_requestor.py,sha256=NT41M0_Utfw61bMqibP1PEdIIQmmKWfzj-Y-xk4
4
4
  seekrai/client.py,sha256=Yhejl-2a-Uoc8nWi-XxETZT4a4Ou_t_TRLcp0e9APIY,5376
5
5
  seekrai/constants.py,sha256=hoR2iF5te5Ydjt_lxIOSGID4vESIakG4F-3xAWdwxaU,1854
6
6
  seekrai/error.py,sha256=rAYL8qEd8INwYMMKvhS-HKeC3QkWL4Wq-zfazFU-zBg,4861
7
- seekrai/filemanager.py,sha256=5tl7fjsLbNy-U1Ko_5iQSdkZ0Bca_bhvR9FhWpaG8FA,9561
7
+ seekrai/filemanager.py,sha256=FAYcjBcRlWzc6-qjTsY4QATSZraegTptV4fYRgTNs5o,9614
8
8
  seekrai/resources/__init__.py,sha256=-OpvrZp0c_ro_e1G14gjOILtcSfpSOeRZeK8kt41WLo,1014
9
9
  seekrai/resources/alignment.py,sha256=MMrADpXNZa0eQ-jqfLcP7-fisJXgpRl-03GPdkjkTcc,5951
10
10
  seekrai/resources/chat/__init__.py,sha256=KmtPupgECtEN80NyvcnSmieTAFXhwmVxhMHP0qhspA4,618
@@ -24,11 +24,11 @@ seekrai/types/alignment.py,sha256=H9rAYsr4rg7uZCgl2tDZjkeq-b_Ov27gljejH8fC26w,19
24
24
  seekrai/types/chat_completions.py,sha256=xRTHBbDJDbz0HgW042WX3csQDolhjEuO81w0rzFSeBU,3691
25
25
  seekrai/types/common.py,sha256=OH3l3u-0_5oz1KYrcHMybFESzivDySocYlJAsLSLOWU,1940
26
26
  seekrai/types/completions.py,sha256=lm9AFdZR3Xg5AHPkV-qETHikkwMJmkHrLGr5GG-YR-M,2171
27
- seekrai/types/deployments.py,sha256=n7_t7DEeBSC8cDJSjIfvWtcgUql1DaEn89zuGYN_RaI,1744
27
+ seekrai/types/deployments.py,sha256=GdZPDaQgzmk9W1aXxZr9CDxqJRNv7NP0pNvqRV3E4xM,1760
28
28
  seekrai/types/embeddings.py,sha256=OANoLNOs0aceS8NppVvvcNYQbF7-pAOAmcr30pw64OU,749
29
29
  seekrai/types/error.py,sha256=uTKISs9aRC4_6zwirtNkanxepN8KY-SqCq0kNbfZylQ,370
30
30
  seekrai/types/files.py,sha256=XmtiM6d9i3tnYS-Kii3QpxZJRqemJi2rvLJ32GsECXQ,2602
31
- seekrai/types/finetune.py,sha256=VNuAZx-Lq4Q55hY47obWnFLDr9if93v5Lwr_qcyCtfU,6124
31
+ seekrai/types/finetune.py,sha256=LUqnLv8oPdasd_F1jGx0WRfNvFDxqO3czl9zfxTxXfg,6140
32
32
  seekrai/types/images.py,sha256=Fusj8OhVYFsT8kz636lRGGivLbPXo_ZNgakKwmzJi3U,914
33
33
  seekrai/types/ingestion.py,sha256=GqDzPjEA_rtiiDKpEmTOXOitZSSKG-JkH1siII5m8oA,152
34
34
  seekrai/types/models.py,sha256=1ZfW9WwayApkISRizDntjkWhYNv-wkbrRVIfHn2QuC4,1242
@@ -39,8 +39,8 @@ seekrai/utils/api_helpers.py,sha256=0Y8BblNIr9h_R12zdmhkxgTlxgoRkbq84QNi4nNWGu8,
39
39
  seekrai/utils/files.py,sha256=7ixn_hgV-6pEhYqLyOp-EN0o8c1CzUwJzX9n3PQ5oqo,7164
40
40
  seekrai/utils/tools.py,sha256=jgJTL-dOIouDbEJLdQpQfpXhqaz_poQYS52adyUtBjo,1781
41
41
  seekrai/version.py,sha256=q6iGQVFor8zXiPP5F-3vy9TndOxKv5JXbaNJ2kdOQws,125
42
- seekrai-0.4.1.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
43
- seekrai-0.4.1.dist-info/METADATA,sha256=28ojGXBP-65YoZv0VfwQ__mN76eC_I53aSso7caRz_E,4748
44
- seekrai-0.4.1.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
45
- seekrai-0.4.1.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
46
- seekrai-0.4.1.dist-info/RECORD,,
42
+ seekrai-0.4.2.dist-info/LICENSE,sha256=xx0jnfkXJvxRnG63LTGOxlggYnIysveWIZ6H3PNdCrQ,11357
43
+ seekrai-0.4.2.dist-info/METADATA,sha256=juc4scPkbrb2iLgXkNTjnf3xw56yJYd1QevX5q_kUlA,4748
44
+ seekrai-0.4.2.dist-info/WHEEL,sha256=sP946D7jFCHeNz5Iq4fL4Lu-PrWrFsgfLXbbkciIZwg,88
45
+ seekrai-0.4.2.dist-info/entry_points.txt,sha256=N49yOEGi1sK7Xr13F_rkkcOxQ88suyiMoOmRhUHTZ_U,48
46
+ seekrai-0.4.2.dist-info/RECORD,,