Skip to content

Instantly share code, notes, and snippets.

@Wikidepia
Created August 27, 2021 22:11
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save Wikidepia/7e6be7dd59df6337ff97f60acf19cf42 to your computer and use it in GitHub Desktop.
Save Wikidepia/7e6be7dd59df6337ff97f60acf19cf42 to your computer and use it in GitHub Desktop.
diff --git a/core.py b/core.py
index 4724d80..e384786 100644
--- a/core.py
+++ b/core.py
@@ -7,7 +7,6 @@
from requests import session, Response
from typing import Optional, Union
from time import sleep
-import numpy as np
import logging
import tarfile
import shutil
@@ -120,8 +119,8 @@ class HybridClient:
else:
data = r.json()
self.shard = data["url"]
- self.start_id = np.int64(data["start_id"])
- self.end_id = np.int64(data["end_id"])
+ self.start_id = (data["start_id"])
+ self.end_id = (data["end_id"])
self.shard_piece = data["shard"]
print("recieved new job")
@@ -293,8 +292,8 @@ class CPUClient:
else:
data = r.json()
self.shard = data["url"]
- self.start_id = np.int64(data["start_id"])
- self.end_id = np.int64(data["end_id"])
+ self.start_id = (data["start_id"])
+ self.end_id = (data["end_id"])
self.shard_piece = data["shard"]
print("recieved new job")
@@ -464,8 +463,8 @@ class GPUClient:
else:
data = r.json()
self.shard = data["url"]
- self.start_id = np.int64(data["start_id"])
- self.end_id = np.int64(data["end_id"])
+ self.start_id = (data["start_id"])
+ self.end_id = (data["end_id"])
self.shard_piece = data["shard"]
print("recieved new job")
diff --git a/requirements.txt b/requirements.txt
index 83c48de..bebaa03 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,2 +1 @@
requests
-numpy
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment