homa 3__tar.gz → 3.2__tar.gz

homa-3.2/PKG-INFO

@@ -0,0 +1,60 @@
+Metadata-Version: 2.1
+Name: homa
+Version: 3.2
+Maintainer: Taha Shieenavaz
+Maintainer-email: tahashieenavaz@gmail.com
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: torchvision
+Requires-Dist: torch
+
+# Homa
+
+- [Homa](#homa)
+  - [Helpers](#helpers)
+    - [Device](#device)
+    - [Pickle](#pickle)
+  - [Transformers](#transformers)
+    - [Positional Encoding](#positional-encoding)
+
+<div align="center">
+    <img src="https://github.com/tahashieenavaz/homa/raw/main/art/homa.svg" width="500" />
+</div>
+
+## Helpers
+
+### Device
+
+```py
+from homa import get_device
+
+torch.tensor(torch.randint(-20, 10, (32, 10))).to(get_device())
+```
+
+### Pickle
+
+```py
+from homa import pickle
+
+person = {"name": "John Doe", age: 88}
+pickle(person, "person.pkl")
+
+loaded_person = pickle("person.pkl")
+```
+
+## Transformers
+
+### Positional Encoding
+
+```py
+from homa import positional_encoding
+
+class TransformerModel(torch.nn.Module):
+    def __init__(self, seq_length, dimension):
+        # ...
+        self.positional_encoding = positional_encoding(seq_length, dimension)
+
+    def forward(self, x):
+        # ...
+        x = x + self.positional_encoding
+```

homa-3.2/README.md

@@ -0,0 +1,50 @@
+# Homa
+
+- [Homa](#homa)
+  - [Helpers](#helpers)
+    - [Device](#device)
+    - [Pickle](#pickle)
+  - [Transformers](#transformers)
+    - [Positional Encoding](#positional-encoding)
+
+<div align="center">
+    <img src="https://github.com/tahashieenavaz/homa/raw/main/art/homa.svg" width="500" />
+</div>
+
+## Helpers
+
+### Device
+
+```py
+from homa import get_device
+
+torch.tensor(torch.randint(-20, 10, (32, 10))).to(get_device())
+```
+
+### Pickle
+
+```py
+from homa import pickle
+
+person = {"name": "John Doe", age: 88}
+pickle(person, "person.pkl")
+
+loaded_person = pickle("person.pkl")
+```
+
+## Transformers
+
+### Positional Encoding
+
+```py
+from homa import positional_encoding
+
+class TransformerModel(torch.nn.Module):
+    def __init__(self, seq_length, dimension):
+        # ...
+        self.positional_encoding = positional_encoding(seq_length, dimension)
+
+    def forward(self, x):
+        # ...
+        x = x + self.positional_encoding
+```

homa-3.2/homa/__init__.py

@@ -0,0 +1,5 @@
+from .helpers import get_device
+from .helpers import pickle
+from .helpers import flush
+
+from .transformers import positional_encoding

{homa-3 → homa-3.2}/homa/helpers.py

@@ -1,22 +1,34 @@
-import pickle
+import pickle as pickle_object
+import torch
 
 
-def pickle(filename_or_variable: str | object, target_file: str | None = None):
+def flush(*args, **kwargs):
+    kwargs["flush"] = True
+    print(*args, **kwargs)
+
+
+def get_device():
+    return torch.device("cuda" if torch.cuda.is_available() else "cpu")
+
+
+def pickle(
+    filename_or_variable: str | object, target_file: str | None = None
+) -> object | 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)
+        return
 
     raise Exception("Wrong pickle helper inputs")
 
 
 def write_to_pickle(data, filename):
     with open(filename, "wb") as f:
-        pickle.dump(data, f)
+        pickle_object.dump(data, f)
 
 
 def read_from_pickle(filename):
     with open(filename, "rb") as f:
-        data = pickle.load(f)
-        return data
+        return pickle_object.load(f)

homa-3.2/homa/transformers.py

@@ -0,0 +1,19 @@
+import torch
+import math
+from .helpers import get_device
+
+
+def positional_encoding(sequence_length: int, model_dimension: int) -> torch.Tensor:
+    positional_embeddings = torch.zeros(
+        sequence_length, model_dimension, device=get_device()
+    )
+    for pos in range(sequence_length):
+        for i in range(0, model_dimension, 2):
+            positional_embeddings[pos, i] = math.sin(
+                pos / (10000 ** (i / model_dimension))
+            )
+            if i + 1 < model_dimension:
+                positional_embeddings[pos, i + 1] = math.cos(
+                    pos / (10000 ** (i / model_dimension))
+                )
+    return positional_embeddings

homa-3.2/homa.egg-info/PKG-INFO

@@ -0,0 +1,60 @@
+Metadata-Version: 2.1
+Name: homa
+Version: 3.2
+Maintainer: Taha Shieenavaz
+Maintainer-email: tahashieenavaz@gmail.com
+Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: torchvision
+Requires-Dist: torch
+
+# Homa
+
+- [Homa](#homa)
+  - [Helpers](#helpers)
+    - [Device](#device)
+    - [Pickle](#pickle)
+  - [Transformers](#transformers)
+    - [Positional Encoding](#positional-encoding)
+
+<div align="center">
+    <img src="https://github.com/tahashieenavaz/homa/raw/main/art/homa.svg" width="500" />
+</div>
+
+## Helpers
+
+### Device
+
+```py
+from homa import get_device
+
+torch.tensor(torch.randint(-20, 10, (32, 10))).to(get_device())
+```
+
+### Pickle
+
+```py
+from homa import pickle
+
+person = {"name": "John Doe", age: 88}
+pickle(person, "person.pkl")
+
+loaded_person = pickle("person.pkl")
+```
+
+## Transformers
+
+### Positional Encoding
+
+```py
+from homa import positional_encoding
+
+class TransformerModel(torch.nn.Module):
+    def __init__(self, seq_length, dimension):
+        # ...
+        self.positional_encoding = positional_encoding(seq_length, dimension)
+
+    def forward(self, x):
+        # ...
+        x = x + self.positional_encoding
+```

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

@@ -4,8 +4,10 @@ setup.py
 homa/__init__.py
 homa/datasets.py
 homa/helpers.py
+homa/transformers.py
 homa.egg-info/PKG-INFO
 homa.egg-info/SOURCES.txt
 homa.egg-info/dependency_links.txt
 homa.egg-info/requires.txt
-homa.egg-info/top_level.txt
+homa.egg-info/top_level.txt
+test/test.py

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

@@ -8,7 +8,7 @@ setup(
     name="homa",
     maintainer="Taha Shieenavaz",
     maintainer_email="tahashieenavaz@gmail.com",
-    version=3,
+    version=3.02,
     packages=find_packages(),
     install_requires=["torchvision", "torch"],
     long_description=description,

homa-3.2/test/test.py

@@ -0,0 +1,4 @@
+from homa import pickle
+
+a = {"name": "Taha", "age": 30}
+pickle(a, "name")

homa-3/PKG-INFO

@@ -1,15 +0,0 @@
-Metadata-Version: 2.1
-Name: homa
-Version: 3
-Maintainer: Taha Shieenavaz
-Maintainer-email: tahashieenavaz@gmail.com
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: torchvision
-Requires-Dist: torch
-
-# Homa
-
-<div align="center">
-    <img src="https://github.com/tahashieenavaz/homa/raw/main/art/homa.svg" width="500" />
-</div>

homa-3/README.md

@@ -1,5 +0,0 @@
-# Homa
-
-<div align="center">
-    <img src="https://github.com/tahashieenavaz/homa/raw/main/art/homa.svg" width="500" />
-</div>

homa-3/homa/__init__.py

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

homa-3/homa.egg-info/PKG-INFO

@@ -1,15 +0,0 @@
-Metadata-Version: 2.1
-Name: homa
-Version: 3
-Maintainer: Taha Shieenavaz
-Maintainer-email: tahashieenavaz@gmail.com
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: torchvision
-Requires-Dist: torch
-
-# Homa
-
-<div align="center">
-    <img src="https://github.com/tahashieenavaz/homa/raw/main/art/homa.svg" width="500" />
-</div>