Issue in Step 1: Prepare source human datasets in Robocasa

[prasadpr09]

Hi,

I am following your documentation (https://mimicgen.github.io/docs/tutorials/getting_started.html) and I am following Option 2 as I want to create my custom dataset in the future.

When I run Step 1: Prepare source human datasets:

python /opt/mimicgen/mimicgen/scripts/download_datasets.py --dataset_type source --tasks square
[robosuite WARNING] No private macro file found! (macros.py:57)
[robosuite WARNING] It is recommended to use a private macro file (macros.py:58)
[robosuite WARNING] To setup, run: python /opt/robosuite/robosuite/scripts/setup_macros.py (macros.py:59)
[robosuite WARNING] Could not import robosuite_models. Some robots may not be available. If you want to use these robots, please install robosuite_models from source (https://github.com/ARISE-Initiative/robosuite_models) or through pip install. (__init__.py:30)
WARNING: robosuite environments not imported...
Got error: No module named 'robosuite.environments.manipulation.single_arm_env'
WARNING: robosuite task zoo environments not imported, possibly because robosuite_task_zoo is not installed...
Got error: No module named 'robosuite.environments.manipulation.single_arm_env'
ROBOMIMIC WARNING(
    No private macro file found!
    It is recommended to use a private macro file
    To setup, run: python /opt/robomimic/robomimic/scripts/setup_macros.py
)

Downloading dataset:
    dataset type: source
    task: square
    download path: /opt/mimicgen/datasets/source/square.hdf5

Downloading square.hdf5: 100%|████████████████████████████████████████████████████████████████████████████████████| 16.5M/16.5M [00:01<00:00, 15.2MB/s]

Then when I try to run :

python /opt/mimicgen/mimicgen/scripts/prepare_src_dataset.py \
--dataset /opt/mimicgen/datasets/source/square.hdf5 \
--env_interface MG_Square \
--env_interface_type robosuite
ROBOMIMIC WARNING(
    No private macro file found!
    It is recommended to use a private macro file
    To setup, run: python /opt/robomimic/robomimic/scripts/setup_macros.py
)
[robosuite WARNING] No private macro file found! (macros.py:57)
[robosuite WARNING] It is recommended to use a private macro file (macros.py:58)
[robosuite WARNING] To setup, run: python /opt/robosuite/robosuite/scripts/setup_macros.py (macros.py:59)
[robosuite WARNING] Could not import robosuite_models. Some robots may not be available. If you want to use these robots, please install robosuite_models from source (https://github.com/ARISE-Initiative/robosuite_models) or through pip install. (__init__.py:30)
WARNING: robosuite environments not imported...
Got error: No module named 'robosuite.environments.manipulation.single_arm_env'
WARNING: robosuite task zoo environments not imported, possibly because robosuite_task_zoo is not installed...
Got error: No module named 'robosuite.environments.manipulation.single_arm_env'

============= Initialized Observation Utils with Obs Spec =============

using obs modality: low_dim with keys: []
using obs modality: rgb with keys: []
Traceback (most recent call last):
  File "/opt/mimicgen/mimicgen/scripts/prepare_src_dataset.py", line 260, in <module>
    prepare_src_dataset(
  File "/opt/mimicgen/mimicgen/scripts/prepare_src_dataset.py", line 135, in prepare_src_dataset
    env = EnvUtils.create_env_for_data_processing(
  File "/opt/robomimic/robomimic/utils/env_utils.py", line 284, in create_env_for_data_processing
    env = env_class.create_for_data_processing(
  File "/opt/robomimic/robomimic/envs/env_robosuite.py", line 434, in create_for_data_processing
    return cls(
  File "/opt/robomimic/robomimic/envs/env_robosuite.py", line 94, in __init__
    self.env = robosuite.make(self._env_name, **kwargs)
  File "/opt/robosuite/robosuite/environments/base.py", line 37, in make
    raise Exception(
Exception: Environment Square_D0 not found. Make sure it is a registered environment among: Lift, Stack, NutAssembly, NutAssemblySingle, NutAssemblySquare, NutAssemblyRound, PickPlace, PickPlaceSingle, PickPlaceMilk, PickPlaceBread, PickPlaceCereal, PickPlaceCan, Door, Wipe, ToolHang, TwoArmLift, TwoArmPegInHole, TwoArmHandover, TwoArmTransport, Kitchen, KitchenDemo, CupcakeCleanup, OrganizeBakingIngredients, PastryDisplay, FillKettle, HeatMultipleWater, VeggieBoil, ArrangeTea, KettleBoiling, PrepareCoffee, ArrangeVegetables, BreadSetupSlicing, ClearingTheCuttingBoard, MeatTransfer, OrganizeVegetables, BowlAndCup, CandleCleanup, ClearingCleaningReceptacles, CondimentCollection, DessertAssembly, DrinkwareConsolidation, FoodCleanup, DefrostByCategory, MicrowaveThawing, QuickThaw, ThawInSink, AssembleCookingArray, FryingPanAdjustment, MealPrepStaging, SearingMeat, SetupFrying, BreadSelection, CheesyBread, PrepareToast, SweetSavoryToastSetup, PrepForTenderizing, PrepMarinatingMeat, ColorfulSalsa, SetupJuicing, SpicyMarinade, HeatMug, MakeLoadedPotato, SimmeringSauce, WaffleReheat, WarmCroissant, BeverageSorting, RestockBowls, RestockPantry, StockingBreakfastFoods, CleanMicrowave, CountertopCleanup, PrepForSanitizing, PushUtensilsToSink, DessertUpgrade, PanTransfer, PlaceFoodInBowls, PrepareSoupServing, ServeSteak, WineServingPrep, ArrangeBreadBasket, BeverageOrganization, DateNight, SeasoningSpiceSetup, SetBowlsForSoup, SizeSorting, BreadAndCheese, CerealAndBowl, MakeFruitBowl, VeggieDipPrep, YogurtDelightPrep, MultistepSteaming, SteamInMicrowave, SteamVegetables, ManipulateDrawer, OpenDrawer, CloseDrawer, DrawerUtensilSort, OrganizeCleaningSupplies, PantryMishap, ShakerShuffle, SnackSorting, DryDishes, DryDrinkware, PreSoakPan, SortingCleanup, StackBowlsInSink, AfterwashSorting, ClearClutter, DrainVeggies, PrewashFoodAssembly, PnPCoffee, CoffeeSetupMug, CoffeeServeMug, CoffeePressButton, ManipulateDoor, OpenDoor, OpenSingleDoor, OpenDoubleDoor, CloseDoor, CloseSingleDoor, CloseDoubleDoor, MicrowavePressButton, TurnOnMicrowave, TurnOffMicrowave, NavigateKitchen, PnP, PnPCounterToCab, PnPCabToCounter, PnPCounterToSink, PnPSinkToCounter, PnPCounterToMicrowave, PnPMicrowaveToCounter, PnPCounterToStove, PnPStoveToCounter, ManipulateSinkFaucet, TurnOnSinkFaucet, TurnOffSinkFaucet, TurnSinkSpout, ManipulateStoveKnob, TurnOnStove, TurnOffStove

How do I fix the Square_D0 error? Any help would be appreciated.

[xbkaishui]

me too, anything update?