[CLEANUP]

pyproject.toml

@@ -4,12 +4,12 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarms-torch"
-version = "0.2.1"
+version = "0.2.2"
 description = "swarms-torch - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <[email protected]>"]
 homepage = "https://github.com/kyegomez/swarms-pytorch"
-documentation = ""  # Add this if you have documentation.
+documentation = "https://github.com/kyegomez/swarms-pytorch"  # Add this if you have documentation.
 readme = "README.md"  # Assuming you have a README.md
 repository = "https://github.com/kyegomez/swarms-pytorch"
 keywords = ["artificial intelligence", "deep learning", "optimizers", "Prompt Engineering"]
@@ -29,10 +29,13 @@ packages = [
 
 [tool.poetry.dependencies]
 python = "^3.6"
-torch = "2.1.2"
-einops = "0.7.0"
-zetascale = "1.4.4"
-pytest = "7.4.2"
+torch = "*"
+einops = "*"
+zetascale = "*"
+pytest = "*"
+torchvision = "*"
+loguru = "*"
+einx = "*"
 
 
 

requirements.txt

@@ -1,8 +1,9 @@
-torch==2.1.2
-einops==0.7.0
-pandas==2.2.1
-zetascale==1.4.4
-pytest==7.4.2
-mkdocs
-mkdocs-material
-mkdocs-glightbox
+python
+torch
+einops
+zetascale
+pytest
+torchvision
+loguru
+einx
+

swarms_torch/ant_colony_swarm.py

@@ -1,7 +1,7 @@
 import torch
+from torch import nn
 
-
-class AntColonyOptimization:
+class AntColonyOptimization(nn.Module):
     """
     Ant Colony Optimization
     Overview: https://en.wikipedia.org/wiki/Ant_colony_optimization_algorithms