fixed install.

.gitignore

@@ -3,6 +3,7 @@ __pycache__/
 *.egg-info/
 logs/
 None/
+dist/
 .pytest_cache/
 .DS_Store
 loco_mujoco/datasets/

loco_mujoco/__init__.py

@@ -1,16 +1,11 @@
 __version__ = '0.1'
 
-
-from .environments import LocoEnv
-
 try:
 
+    from .environments import LocoEnv
+
     def get_all_task_names():
         return LocoEnv.get_all_task_names()
 
-
-    def download_all_datasets():
-        return LocoEnv.download_all_datasets()
-
-except ImportError:
-    pass
+except ImportError as e:
+    print(e)

loco_mujoco/environments/base.py

@@ -1,7 +1,5 @@
 import os
 
-import wget
-import zipfile
 import warnings
 from pathlib import Path
 from copy import deepcopy
@@ -953,36 +951,6 @@ def get_all_task_names(cls):
 
     _registered_envs = dict()
 
-    @classmethod
-    def download_all_datasets(cls):
-        """
-        Download and installs all datasets.
-
-        """
-        dataset_path = Path(loco_mujoco.__file__).resolve().parent / "datasets"
-
-        print("Downloading Humanoid Datasets ...\n")
-        dataset_path_humanoid = dataset_path / "humanoids/real"
-        dataset_path_humanoid_str = str(dataset_path_humanoid)
-        humanoid_url = "https://zenodo.org/records/10102870/files/humanoid_datasets_v0.1.zip?download=1"
-        wget.download(humanoid_url, out=dataset_path_humanoid_str)
-        file_name = "humanoid_datasets_v0.1.zip"
-        file_path = str(dataset_path_humanoid / file_name)
-        with zipfile.ZipFile(file_path, "r") as zip_ref:
-            zip_ref.extractall(dataset_path_humanoid_str)
-        os.remove(file_path)
-
-        print("Downloading Quadruped Datasets ...\n")
-        dataset_path_quadrupeds = dataset_path / "quadrupeds/real"
-        dataset_path_quadrupeds_str = str(dataset_path_quadrupeds)
-        quadruped_url = "https://zenodo.org/records/10102870/files/quadruped_datasets_v0.1.zip?download=1"
-        wget.download(quadruped_url, out=dataset_path_quadrupeds_str)
-        file_name = "quadruped_datasets_v0.1.zip"
-        file_path = str(dataset_path_quadrupeds / file_name)
-        with zipfile.ZipFile(file_path, "r") as zip_ref:
-            zip_ref.extractall(dataset_path_quadrupeds_str)
-        os.remove(file_path)
-
 
 class ValidTaskConf:
 

loco_mujoco/utils/__init__.py

@@ -3,3 +3,4 @@
 from .checks import *
 from .video import video2gif
 from .domain_randomization import *
+from .dataset import download_all_datasets, download_real_datasets, download_perfect_dataset

loco_mujoco/utils/dataset.py

@@ -0,0 +1,82 @@
+import os
+import wget
+import zipfile
+from pathlib import Path
+import loco_mujoco
+
+
+def download_all_datasets():
+    """
+    Download and installs all datasets.
+
+    """
+    download_real_datasets()
+    download_perfect_dataset()
+
+
+def download_real_datasets():
+    """
+    Download and installs real datasets.
+
+    """
+
+    dataset_path = Path(loco_mujoco.__file__).resolve().parent / "datasets"
+    print(dataset_path)
+
+    print("Downloading Humanoid Datasets ...\n")
+    dataset_path_humanoid = dataset_path / "humanoids/real"
+    dataset_path_humanoid_str = str(dataset_path_humanoid)
+    os.makedirs(dataset_path_humanoid_str, exist_ok=True)
+    humanoid_url = "https://zenodo.org/records/10102870/files/humanoid_datasets_v0.1.zip?download=1"
+    wget.download(humanoid_url, out=dataset_path_humanoid_str)
+    file_name = "humanoid_datasets_v0.1.zip"
+    file_path = str(dataset_path_humanoid / file_name)
+    with zipfile.ZipFile(file_path, "r") as zip_ref:
+        zip_ref.extractall(dataset_path_humanoid_str)
+    os.remove(file_path)
+
+    print("Downloading Quadruped Datasets ...\n")
+    dataset_path_quadrupeds = dataset_path / "quadrupeds/real"
+    dataset_path_quadrupeds_str = str(dataset_path_quadrupeds)
+    os.makedirs(dataset_path_quadrupeds_str, exist_ok=True)
+    quadruped_url = "https://zenodo.org/records/10102870/files/quadruped_datasets_v0.1.zip?download=1"
+    wget.download(quadruped_url, out=dataset_path_quadrupeds_str)
+    file_name = "quadruped_datasets_v0.1.zip"
+    file_path = str(dataset_path_quadrupeds / file_name)
+    with zipfile.ZipFile(file_path, "r") as zip_ref:
+        zip_ref.extractall(dataset_path_quadrupeds_str)
+    os.remove(file_path)
+
+
+def download_perfect_dataset():
+    """
+    Download and installs perfect datasets.
+
+    """
+    dataset_path = Path(loco_mujoco.__file__).resolve().parent / "datasets"
+
+    print("Downloading Perfect Humanoid Datasets ...\n")
+    dataset_path_humanoid = dataset_path / "humanoids/perfect"
+    dataset_path_humanoid_str = str(dataset_path_humanoid)
+    os.makedirs(dataset_path_humanoid_str, exist_ok=True)
+    humanoid_url = "https://zenodo.org/records/10102870/files/humanoid_datasets_v0.1.zip?download=1"
+    wget.download(humanoid_url, out=dataset_path_humanoid_str)
+    file_name = "humanoid_datasets_v0.1.zip"
+    file_path = str(dataset_path_humanoid / file_name)
+    with zipfile.ZipFile(file_path, "r") as zip_ref:
+        zip_ref.extractall(dataset_path_humanoid_str)
+    os.remove(file_path)
+
+
+    print("Downloading Perfect Quadruped Datasets ...\n")
+    dataset_path_quadrupeds = dataset_path / "quadrupeds/perfect"
+    dataset_path_quadrupeds_str = str(dataset_path_quadrupeds)
+    os.makedirs(dataset_path_quadrupeds_str, exist_ok=True)
+    quadruped_url = "https://zenodo.org/records/10102870/files/quadruped_datasets_v0.1.zip?download=1"
+    wget.download(quadruped_url, out=dataset_path_quadrupeds_str)
+    file_name = "quadruped_datasets_v0.1.zip"
+    file_path = str(dataset_path_quadrupeds / file_name)
+    with zipfile.ZipFile(file_path, "r") as zip_ref:
+        zip_ref.extractall(dataset_path_quadrupeds_str)
+    os.remove(file_path)
+

pyproject.toml

@@ -1,5 +1,5 @@
 [build-system]
-requires = ["setuptools", "wheel"]
+requires = ["setuptools", "wheel", "wget"]
 
 [project]
 name = "loco-mujoco"
@@ -35,4 +35,6 @@ Repository = "https://github.com/robfiras/loco-mujoco"
 Issues = "https://github.com/robfiras/loco-mujoco/issues"
 
 [project.scripts]
-loco-mujoco-download = "loco_mujoco:download_all_datasets"
+loco-mujoco-download = "loco_mujoco.utils:download_all_datasets"
+loco-mujoco-download-real = "loco_mujoco.utils:download_real_datasets"
+loco-mujoco-download-perfect = "loco_mujoco.utils:download_perfect_datasets"

setup.py

@@ -1,5 +1,4 @@
 from setuptools import setup, find_packages
-from os import path
 import glob
 from loco_mujoco import __version__