Cannot execute the code

[Kunal-Kumar-Sahoo]

While executing the following command:

CUDA_VISIBLE_DEVICES=0 python -m JaxPref.new_preference_reward_main --use_human_label True --comment test_1 --transformer.embd_dim 256 --transformer.n_layer 1 --transformer.n_head 4 --env kitchen --logging.output_dir './logs/pref_reward' --batch_size 256 --num_query 50 --query_len 100 --n_epochs 10000 --skip_flag 0 --seed 42 --model_type PrefTransformer

I get the following error:

/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/site-packages/Cython/Distutils/old_build_ext.py:15: DeprecationWarning: dep_util is Deprecated. Use functions from setuptools instead.
  from distutils.dep_util import newer, newer_group
/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/site-packages/llvmlite/binding/ffi.py:175: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html
  from pkg_resources import resource_filename
/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/site-packages/pkg_resources/__init__.py:3117: DeprecationWarning: Deprecated call to `pkg_resources.declare_namespace('mpl_toolkits')`.
Implementing implicit namespace packages (as specified in PEP 420) is preferred to `pkg_resources.declare_namespace`. See https://setuptools.pypa.io/en/latest/references/keywords.html#keyword-namespace-packages
  declare_namespace(pkg)
/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/site-packages/pkg_resources/__init__.py:3117: DeprecationWarning: Deprecated call to `pkg_resources.declare_namespace('google')`.
Implementing implicit namespace packages (as specified in PEP 420) is preferred to `pkg_resources.declare_namespace`. See https://setuptools.pypa.io/en/latest/references/keywords.html#keyword-namespace-packages
  declare_namespace(pkg)
Traceback (most recent call last):
  File "/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/runpy.py", line 194, in _run_module_as_main
    return _run_code(code, main_globals, None,
  File "/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/runpy.py", line 87, in _run_code
    exec(code, run_globals)
  File "/home/rankit/PreferenceTransformer/JaxPref/new_preference_reward_main.py", line 18, in <module>
    import robosuite as suite
  File "/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/site-packages/robosuite/__init__.py", line 4, in <module>
    from robosuite.environments.manipulation.lift import Lift
  File "/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/site-packages/robosuite/environments/manipulation/lift.py", line 5, in <module>
    from robosuite.environments.manipulation.single_arm_env import SingleArmEnv
  File "/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/site-packages/robosuite/environments/manipulation/single_arm_env.py", line 3, in <module>
    from robosuite.environments.manipulation.manipulation_env import ManipulationEnv
  File "/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/site-packages/robosuite/environments/manipulation/manipulation_env.py", line 3, in <module>
    from robosuite.environments.robot_env import RobotEnv
  File "/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/site-packages/robosuite/environments/robot_env.py", line 7, in <module>
    from robosuite.controllers import reset_controllers
  File "/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/site-packages/robosuite/controllers/__init__.py", line 1, in <module>
    from .controller_factory import controller_factory, load_controller_config, reset_controllers, get_pybullet_server
  File "/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/site-packages/robosuite/controllers/controller_factory.py", line 10, in <module>
    from .interpolators.linear_interpolator import LinearInterpolator
  File "/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/site-packages/robosuite/controllers/interpolators/linear_interpolator.py", line 3, in <module>
    import robosuite.utils.transform_utils as T
  File "/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/site-packages/robosuite/utils/transform_utils.py", line 11, in <module>
    from robosuite.utils.numba import jit_decorator
  File "/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/site-packages/robosuite/utils/numba.py", line 4, in <module>
    import numba
  File "/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/site-packages/numba/__init__.py", line 43, in <module>
    from numba.np.ufunc import (vectorize, guvectorize, threading_layer,
  File "/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/site-packages/numba/np/ufunc/__init__.py", line 3, in <module>
    from numba.np.ufunc.decorators import Vectorize, GUVectorize, vectorize, guvectorize
  File "/home/rankit/micromamba/envs/preference_transformer/lib/python3.8/site-packages/numba/np/ufunc/decorators.py", line 3, in <module>
    from numba.np.ufunc import _internal
SystemError: initialization of _internal failed without raising an exception

Pl help me figure this out!