work

.gitignore

@@ -1,4 +1,7 @@
 
 demos/
 data/
-exps/
+exps/
+
+__pycache__
+*egg-info

pyproject.toml

@@ -0,0 +1,89 @@
+# Package ######################################################################
+
+[build-system]
+requires = ["setuptools >= 62.0.0"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "rfcl"
+description = "Reverse Forward Curriculum Learning"
+readme = "README.md"
+requires-python = ">= 3.8"
+authors = [{ name = "Anon", email = "[email protected]" }]
+license = { text = "MIT License" }
+keywords = ["Reinforcement Learning", "Robotics", "Jax", "RL", "AI"]
+classifiers = [
+    "Development Status :: 4 - Beta",  # change to `5 - Production/Stable` when ready
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3",
+    "Programming Language :: Python :: 3.8",
+    "Programming Language :: Python :: 3.9",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    'Intended Audience :: Science/Research',
+    'Topic :: Scientific/Engineering :: Artificial Intelligence',
+]
+dependencies = [
+    "numpy >=1.22.0",
+    "scipy",
+
+    # RL
+    "gymnasium",
+    "gymnax",
+
+    # Jax Related
+    "chex",
+    "distrax",
+    "flax",
+    "optax",
+
+    # Video generation
+    "moviepy",
+
+    # Exp tracking
+    "tensorboard",
+    "tensorboardX",
+    "wandb",
+    "omegaconf",
+
+    # Vision
+    "opencv-python",
+    "transforms3d",
+
+    # Other
+    "tqdm",
+    "dacite",
+    "matplotlib"
+]
+dynamic = ["version"]
+
+[project.optional-dependencies]
+
+mani-skill2 = ["mani_skill2"]
+dm-control = ["dm_control"]
+brax = ["brax", "gymnax"]
+gymnax = ["gymnax"]
+
+dev = ["ipdb", "jupyterlab", "ipython", "black", "isort"]
+
+[tool.setuptools]
+include-package-data = true
+
+[tool.setuptools.packages.find]
+include = ["rfcl", "rfcl.*"]
+
+# [tool.setuptools.package-data]
+
+# Linters and Test tools #######################################################
+
+[tool.black]
+safe = true
+
+[tool.isort]
+atomic = true
+profile = "black"
+src_paths = ["rfcl", "scripts"]
+extra_standard_library = ["typing_extensions"]
+indent = 4
+lines_after_imports = 2
+multi_line_output = 3