Moving virtual env to Pixi + Makefile

.gitattributes

@@ -0,0 +1,2 @@
+# SCM syntax highlighting & preventing 3-way merges
+pixi.lock merge=binary linguist-language=YAML linguist-generated=true

.gitignore

@@ -23,3 +23,6 @@ examples/*
 !examples/goldfish.jpg
 checkpoints
 utils/hugging_face/
+# pixi environments
+.pixi/*
+!.pixi/config.toml

Makefile

@@ -0,0 +1,16 @@
+.PHONY: setup train-image train-maze analysis-image clean
+
+setup:
+	pixi install
+
+train-image:
+	pixi run python -m tasks.image_classification.train
+
+train-maze:
+	pixi run python -m tasks.mazes.train
+
+analysis-image:
+	pixi run python tasks/image_classification/analysis/run_imagenet_analysis.py
+
+clean:
+	rm -rf .pixi

pixi.toml

@@ -0,0 +1,29 @@
+[workspace]
+authors = ["Uday Phalak <udayphlk@gmail.com>"]
+channels = ["pytorch", "conda-forge"]
+name = "continuous-thought-machines"
+platforms = ["osx-arm64"]
+version = "0.1.0"
+
+[dependencies]
+python = "3.12.*"
+pytorch = "*"
+torchvision = "*"
+numpy = "*"
+matplotlib = "*"
+seaborn = "*"
+tqdm = "*"
+opencv = "*"
+imageio = "*"
+scikit-learn = "*"
+umap-learn = "*"
+python-dotenv = "*"
+gymnasium = "*"
+datasets = "*"
+huggingface_hub = "*"
+safetensors = "*"
+ffmpeg = "*"
+
+[pypi-dependencies]
+autoclip = "*"
+minigrid = "*"