Update NutAssembly to handle different mounts aside from default

Author: kevin-thankyou-lin

robosuite/environments/manipulation/nut_assembly.py

@@ -159,6 +159,7 @@ def __init__(
         env_configuration="default",
         controller_configs=None,
         gripper_types="default",
+        base_types="default",
         initialization_noise="default",
         table_full_size=(0.8, 0.8, 0.05),
         table_friction=(1, 0.005, 0.0001),
@@ -218,7 +219,7 @@ def __init__(
             robots=robots,
             env_configuration=env_configuration,
             controller_configs=controller_configs,
-            base_types="default",
+            base_types=base_types,
             gripper_types=gripper_types,
             initialization_noise=initialization_noise,
             use_camera_obs=use_camera_obs,