Add table_offset as an arg for NutAssembly

Author: kevin-thankyou-lin

robosuite/environments/manipulation/nut_assembly.py

@@ -163,6 +163,7 @@ def __init__(
         initialization_noise="default",
         table_full_size=(0.8, 0.8, 0.05),
         table_friction=(1, 0.005, 0.0001),
+        table_offset=(0, 0, 0.82),
         use_camera_obs=True,
         use_object_obs=True,
         reward_scale=1.0,
@@ -203,7 +204,7 @@ def __init__(
         # settings for table top
         self.table_full_size = table_full_size
         self.table_friction = table_friction
-        self.table_offset = np.array((0, 0, 0.82))
+        self.table_offset = table_offset
 
         # reward configuration
         self.reward_scale = reward_scale