homa 2.2__tar.gz → 3__tar.gz

{homa-2.2 → homa-3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: homa
-Version: 2.2
+Version: 3
 Maintainer: Taha Shieenavaz
 Maintainer-email: tahashieenavaz@gmail.com
 Description-Content-Type: text/markdown

{homa-2.2 → homa-3}/homa/__init__.py

@@ -1,2 +1,3 @@
 from .datasets import ImageDataset
 from .datasets import AugmentedDataset
+from .helpers import pickle

homa-3/homa/helpers.py

@@ -0,0 +1,22 @@
+import pickle
+
+
+def pickle(filename_or_variable: str | object, target_file: str | None = None):
+    if not target_file and isinstance(filename_or_variable, str):
+        return read_from_pickle(filename_or_variable)
+
+    if target_file and isinstance(filename_or_variable, object):
+        write_to_pickle(filename_or_variable, target_file)
+
+    raise Exception("Wrong pickle helper inputs")
+
+
+def write_to_pickle(data, filename):
+    with open(filename, "wb") as f:
+        pickle.dump(data, f)
+
+
+def read_from_pickle(filename):
+    with open(filename, "rb") as f:
+        data = pickle.load(f)
+        return data

{homa-2.2 → homa-3}/homa.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: homa
-Version: 2.2
+Version: 3
 Maintainer: Taha Shieenavaz
 Maintainer-email: tahashieenavaz@gmail.com
 Description-Content-Type: text/markdown

{homa-2.2 → homa-3}/homa.egg-info/SOURCES.txt

@@ -3,6 +3,7 @@ README.md
 setup.py
 homa/__init__.py
 homa/datasets.py
+homa/helpers.py
 homa.egg-info/PKG-INFO
 homa.egg-info/SOURCES.txt
 homa.egg-info/dependency_links.txt

{homa-2.2 → homa-3}/setup.py

@@ -1,22 +1,14 @@
 from setuptools import setup
 from setuptools import find_packages
-from sys import argv
 
 with open("README.md") as fh:
     description = fh.read()
 
-with open("version.txt", "r") as fh:
-    current_version = float(fh.readline())
-
-with open("version.txt", "w") as fh:
-    next_version = round(current_version + 0.01, 2)
-    fh.write(str(next_version))
-
 setup(
     name="homa",
     maintainer="Taha Shieenavaz",
     maintainer_email="tahashieenavaz@gmail.com",
-    version=next_version,
+    version=3,
     packages=find_packages(),
     install_requires=["torchvision", "torch"],
     long_description=description,