{
  "format": "rlds",
  "dataset_name": "Flexa-pick-mug-v5",
  "num_steps": 270,
  "steps": [
    {
      "observation": {
        "qpos_arm": [
          5.3091071608311e-05,
          0.3043766886928164,
          -2.767955234796154e-05,
          -1.574456560682767,
          -0.0006763619708380207,
          1.9995620314665616,
          -0.7853048005439325
        ],
        "ee_pos": [
          0.6767310509486302,
          -7.483338798507821e-05,
          0.3775715929814632
        ],
        "mug_pos": [
          0.5,
          0.0,
          0.29029119999999997
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0,
        0.3,
        0.0,
        -1.57079,
        0.0,
        2.0,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": true,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.00010701777980025957,
          0.31409877356766586,
          -7.100103893507822e-05,
          -1.5818979571289216,
          -0.0018318415715377007,
          1.99561404639093,
          -0.7853128347314403
        ],
        "ee_pos": [
          0.6746421464553206,
          -0.00024170657033413182,
          0.3670731848216978
        ],
        "mug_pos": [
          0.5000087056078663,
          2.08689064088784e-05,
          0.2878368487038409
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0,
        0.3,
        0.0,
        -1.57079,
        0.0,
        2.0,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.00011908982493965296,
          0.3253244811701806,
          -0.00015260873390598399,
          -1.5903804797697816,
          -0.0023802722020764304,
          1.9909171297056665,
          -0.7853116831469498
        ],
        "ee_pos": [
          0.6719192385951722,
          -0.0003739387537705737,
          0.3548050401389508
        ],
        "mug_pos": [
          0.5000057959035855,
          -6.647655821722079e-06,
          0.28874424700874957
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0,
        0.3,
        0.0,
        -1.57079,
        0.0,
        2.0,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          8.798193715583618e-05,
          0.3353858059351673,
          -0.0002615437013462073,
          -1.597904300841005,
          -0.0024547795607329946,
          1.9886638409605577,
          -0.7853002514075359
        ],
        "ee_pos": [
          0.6696466045702997,
          -0.00047561250404290945,
          0.3439810146308139
        ],
        "mug_pos": [
          0.5000228792355806,
          7.87321859625063e-06,
          0.28904130483404483
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0,
        0.3,
        0.0,
        -1.57079,
        0.0,
        2.0,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          1.5814495818346522e-05,
          0.3423643053579093,
          -0.00038861410104376555,
          -1.603720801218562,
          -0.0025325620438044617,
          1.9876428974232243,
          -0.78529527188274
        ],
        "ee_pos": [
          0.6679255185512749,
          -0.0006127411977317142,
          0.33617872362383305
        ],
        "mug_pos": [
          0.5000229069324306,
          1.9137531835927718e-06,
          0.28907804911731155
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.00011542839495696948,
        0.29874854433561554,
        -0.00010780999508220537,
        -1.5720686663546788,
        -2.5834353861544864e-05,
        1.9985234991150884,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -8.865410660965557e-05,
          0.3456653506674976,
          -0.0005271840169959626,
          -1.6076065714012728,
          -0.0026162788451847873,
          1.9860755247269044,
          -0.7852982463194292
        ],
        "ee_pos": [
          0.6665406210506406,
          -0.0007823702225848699,
          0.3317462614853938
        ],
        "mug_pos": [
          0.49998178024708023,
          1.8134164888032542e-05,
          0.28907591395139237
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.00020744584865826752,
        0.29719465661830324,
        -0.00019384163613436864,
        -1.5736253408119352,
        -4.64778698627687e-05,
        1.996713123542173,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.0002218577377979058,
          0.34540252835561924,
          -0.0006705696279671868,
          -1.6102170073586755,
          -0.00256057391935144,
          1.983462918688272,
          -0.7853008195231788
        ],
        "ee_pos": [
          0.665294659203557,
          -0.0009529992108388449,
          0.3302586852389566
        ],
        "mug_pos": [
          0.5000272114118298,
          1.2915549580910224e-05,
          0.2890740014491183
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.00038300338848188206,
        0.2942422434673897,
        -0.00035821568418222465,
        -1.5764701516142134,
        -8.603084020316625e-05,
        1.993357074051667,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.00036461522572876586,
          0.34250689031734227,
          -0.0008076239369948335,
          -1.611941620620305,
          -0.002411594529359918,
          1.9803715468200178,
          -0.7852987306898036
        ],
        "ee_pos": [
          0.6641860416974382,
          -0.0011123261261841541,
          0.33095697204885266
        ],
        "mug_pos": [
          0.5000242091300388,
          1.094533691014698e-05,
          0.2890713545140791
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.00047411470368555913,
        0.2912832537158828,
        -0.00044382643293268685,
        -1.5792831838477739,
        -0.00010666119244885862,
        1.9900195181661955,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.0004886168334018547,
          0.3380790368002968,
          -0.0009184199632470064,
          -1.6131819217408105,
          -0.00227235362673066,
          1.9775776688267601,
          -0.7852962776084756
        ],
        "ee_pos": [
          0.6632729464410461,
          -0.0012447016584316837,
          0.3329840859007014
        ],
        "mug_pos": [
          0.4999931144623252,
          2.348278625512041e-05,
          0.28907278134971953
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0005080450475334905,
        0.28818566299176557,
        -0.00047601791286734396,
        -1.582203654000274,
        -0.00011436801212748344,
        1.9865408992834004,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.0005763193360608958,
          0.3329720047791779,
          -0.0009897280775565452,
          -1.6143747624755553,
          -0.0021932782801987296,
          1.974736630978162,
          -0.7852971671962473
        ],
        "ee_pos": [
          0.6623695854905597,
          -0.001337870318622688,
          0.3355211483446311
        ],
        "mug_pos": [
          0.5000204608224561,
          1.1637901591033702e-05,
          0.28907372290783706
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0005293243130210456,
        0.284892487185509,
        -0.0004964239325490934,
        -1.5852869421621734,
        -0.00011921560675020903,
        1.982855433798472,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.0006231051665233001,
          0.3277563277952965,
          -0.001019836279932432,
          -1.6159040287018707,
          -0.002150901678701826,
          1.9712929173077967,
          -0.7852992230290688
        ],
        "ee_pos": [
          0.6612516534564644,
          -0.0013838647945477404,
          0.33793648623508366
        ],
        "mug_pos": [
          0.5000273465768603,
          8.422499398545438e-06,
          0.2890715858522675
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0005340309660124113,
        0.2813815484777694,
        -0.000501330500173175,
        -1.5885523119246447,
        -0.00012026483966150733,
        1.9789387375503433,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.000633983271255557,
          0.32276274674195077,
          -0.0010148211997173609,
          -1.6180179435982889,
          -0.002122504642218619,
          1.967193018065036,
          -0.7853000277520262
        ],
        "ee_pos": [
          0.6598358296098549,
          -0.0013854565136927504,
          0.3398693489050139
        ],
        "mug_pos": [
          0.4999897364185983,
          2.836523233898761e-05,
          0.28907314166354503
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.000535110035648377,
        0.27764023268752397,
        -0.0005028623447059381,
        -1.592010967604912,
        -0.0001204751031600061,
        1.974776318662588,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.000618500079288627,
          0.3177447140153273,
          -0.0009881964490978245,
          -1.6212297476418551,
          -0.0021193484894652535,
          1.961736065384593,
          -0.7853008542211276
        ],
        "ee_pos": [
          0.6578652996771112,
          -0.0013588056263289124,
          0.3411619205775055
        ],
        "mug_pos": [
          0.5000199073048142,
          5.466820208859183e-06,
          0.28907320342090354
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0005131664189310014,
        0.27237417099838857,
        -0.0004829139480079734,
        -1.5968368196758227,
        -0.0001152755155117897,
        1.9689413783502006,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.0005775784810351476,
          0.3131739482058037,
          -0.0009382852613592975,
          -1.624800234135679,
          -0.0021224700851361594,
          1.9563637263590647,
          -0.7852992172215468
        ],
        "ee_pos": [
          0.6557580473585597,
          -0.0013028909494757982,
          0.34197496853246645
        ],
        "mug_pos": [
          0.500022637354689,
          6.274106628459097e-06,
          0.28907201273379735
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0004457529370877011,
        0.2687639970908602,
        -0.00041981125933880496,
        -1.600123009095825,
        -9.930467054706107e-05,
        1.9649524687226483,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.0005142880917383101,
          0.30866497437395213,
          -0.0008728126906739766,
          -1.629018191159142,
          -0.0021223303105962167,
          1.9512071344451893,
          -0.7852992634915296
        ],
        "ee_pos": [
          0.6535542730934648,
          -0.0012212541236359697,
          0.3424514062061975
        ],
        "mug_pos": [
          0.49999028499252884,
          2.3313494058809508e-05,
          0.28907239945085483
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0003560846345404483,
        0.2633530192171266,
        -0.00033572048005159024,
        -1.6050268764093898,
        -7.789287063671602e-05,
        1.958981649050509,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.00043530729028554523,
          0.3044041347424999,
          -0.0007901962827881512,
          -1.6331899012257625,
          -0.002136058713917692,
          1.9461838592593537,
          -0.7852995263100482
        ],
        "ee_pos": [
          0.6513509125674922,
          -0.0011214664749780836,
          0.3427843529235591
        ],
        "mug_pos": [
          0.5000201224767703,
          7.484683276630054e-06,
          0.28907314610895746
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.00029368289305764197,
        0.2595481417117077,
        -0.00027704531390720987,
        -1.6084503575089337,
        -6.286920839658032e-05,
        1.9547958412683775,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.00034607038085761663,
          0.29985668371358826,
          -0.0007028473520560787,
          -1.6377948575247236,
          -0.002130985917885734,
          1.940666944160091,
          -0.785300630127208
        ],
        "ee_pos": [
          0.6489639436657496,
          -0.0010090497296303194,
          0.3430531875770537
        ],
        "mug_pos": [
          0.5000225275938858,
          4.485832748376873e-06,
          0.2890703251493007
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0001830143096488117,
        0.25373433356400127,
        -0.00017263598684768568,
        -1.613675360787845,
        -3.601686174355576e-05,
        1.948394747591677,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.00025283341770545773,
          0.2948670255553117,
          -0.000609611288078507,
          -1.642725686660484,
          -0.002122264128565912,
          1.9339905207662538,
          -0.7853007855908711
        ],
        "ee_pos": [
          0.6462230441884366,
          -0.0008904195934516914,
          0.3433439269950045
        ],
        "mug_pos": [
          0.4999850126593785,
          1.649682449692802e-05,
          0.28907411357214485
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.00010673156946219542,
        0.24814090067758773,
        -0.00010050475289845423,
        -1.618692056333972,
        -1.7321053594613863e-05,
        1.9422326821715785,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.000161514626542482,
          0.2895626873087694,
          -0.0005157754931333816,
          -1.6477866259499685,
          -0.002097367197331386,
          1.9273798100370882,
          -0.7852997638618013
        ],
        "ee_pos": [
          0.643432221178253,
          -0.0007702519516919571,
          0.3437773558272146
        ],
        "mug_pos": [
          0.5000283894849796,
          1.0449002441063908e-05,
          0.2890742830014609
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.3408839968549815e-05,
        0.2425404963557584,
        -4.0467878394251864e-05,
        -1.6237007795242728,
        -1.6536762330089015e-06,
        1.9360625984231532,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -7.736426097088345e-05,
          0.28402608016005576,
          -0.0004276468659047859,
          -1.6528419277902497,
          -0.002079213181050782,
          1.921270540381083,
          -0.7852996115169839
        ],
        "ee_pos": [
          0.6407363387174981,
          -0.0006592328726749115,
          0.3444050551973733
        ],
        "mug_pos": [
          0.5000220454085555,
          7.1923979816722826e-06,
          0.2890717152819354
        ],
        "gripper_state": 1.0
      },
      "action": [
        7.625724827188635e-06,
        0.23688334775635891,
        8.053216973305583e-06,
        -1.6287337910121307,
        1.1090576351214024e-05,
        1.9298387930173402,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0873426840307481e-06,
          0.2782643809259305,
          -0.00034849165910481845,
          -1.6578652591964391,
          -0.002069717107398966,
          1.9151861980446423,
          -0.7853003373046886
        ],
        "ee_pos": [
          0.6380514653037347,
          -0.0005610857576436763,
          0.3451951726178425
        ],
        "mug_pos": [
          0.4999889861184328,
          2.2436099555137368e-05,
          0.28907430800508926
        ],
        "gripper_state": 1.0
      },
      "action": [
        7.854759723555117e-05,
        0.2310928216723169,
        7.567709664821265e-05,
        -1.6338689968409659,
        2.897625604768652e-05,
        1.9234689635237043,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          6.681193952309303e-05,
          0.2722611352734416,
          -0.0002773735964746398,
          -1.6629386276457578,
          -0.0020525079040341664,
          1.9086988370354003,
          -0.7853007150327973
        ],
        "ee_pos": [
          0.6352621667413512,
          -0.0004727025595838044,
          0.3460824661398485
        ],
        "mug_pos": [
          0.5000264467130271,
          1.2283298856929945e-05,
          0.28907506463700355
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.00013513115691607728,
        0.22508721788347733,
        0.00012981430747498242,
        -1.6391933031280141,
        4.337436407830315e-05,
        1.916851128694659,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.00012841743166266346,
          0.2660161419737154,
          -0.00021272782368496606,
          -1.6681640494326713,
          -0.0020262018852208264,
          1.9017781566566714,
          -0.785300304661356
        ],
        "ee_pos": [
          0.632329101186475,
          -0.00039162544482329565,
          0.347014749754709
        ],
        "mug_pos": [
          0.5000275964913891,
          9.297918608632457e-06,
          0.28907147339910805
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0002014382267510771,
        0.21882671855138922,
        0.0001934199843418293,
        -1.6447475741689421,
        6.0420027915939906e-05,
        1.9099356144337212,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.00018799475197143545,
          0.2595484118039307,
          -0.00015030086103012596,
          -1.6735938930253835,
          -0.002001533282789915,
          1.894562747211674,
          -0.7852996683108137
        ],
        "ee_pos": [
          0.6292649698230608,
          -0.00031432559966275517,
          0.34796521631375116
        ],
        "mug_pos": [
          0.4999932391049556,
          2.8054462333238665e-05,
          0.2890724239840914
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.00027403957123663575,
        0.21232989372944236,
        0.00026325727952874334,
        -1.6505094377970446,
        7.927790284803014e-05,
        1.9027456956817672,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.0002494033809308819,
          0.25249415394396196,
          -8.827800785579608e-05,
          -1.6796778009090965,
          -0.0019866308447617953,
          1.8861603635502915,
          -0.7853001829665301
        ],
        "ee_pos": [
          0.625782644965073,
          -0.00023890458735065755,
          0.3488833798280415
        ],
        "mug_pos": [
          0.5000200436572849,
          9.218043123281371e-06,
          0.28907326770824293
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.00036626426469630604,
        0.20426928635731612,
        0.00035225563546303034,
        -1.6576592249795365,
        0.00010350679742919416,
        1.893802866552808,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.00031301920406238254,
          0.24537652121711295,
          -1.770144442368126e-05,
          -1.6857350502368265,
          -0.0019719193019126437,
          1.8778809301563666,
          -0.7852988570559764
        ],
        "ee_pos": [
          0.622263628369848,
          -0.00015807137079356252,
          0.34985213320713043
        ],
        "mug_pos": [
          0.5000222811071761,
          4.277728366182543e-06,
          0.28907103690683134
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.00040388194818457163,
        0.19797771659860677,
        0.0003888935296272835,
        -1.663244092325332,
        0.00011344083197198852,
        1.8868029595565619,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.0003674914821524934,
          0.23792338149386522,
          4.5645794463778226e-05,
          -1.6921936204819252,
          -0.0019607202291426363,
          1.8697506924369378,
          -0.7852998332261564
        ],
        "ee_pos": [
          0.6186929440932675,
          -8.655517213077319e-05,
          0.35085560204873123
        ],
        "mug_pos": [
          0.4999875726237886,
          1.8757145973807583e-05,
          0.28907393221072947
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0004071201257581971,
        0.1897219332732289,
        0.00039266222748432484,
        -1.6705720700368156,
        0.00011412191947188963,
        1.8775959979661383,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.00040904720099493337,
          0.2304877315374215,
          9.85162895141228e-05,
          -1.698489182137204,
          -0.0019720635864358812,
          1.8616674957255253,
          -0.7853001909999915
        ],
        "ee_pos": [
          0.6151263993926207,
          -3.4387791391322356e-05,
          0.35191804833631773
        ],
        "mug_pos": [
          0.5000224738554646,
          3.659417007252129e-06,
          0.28907334502892373
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.00041184258432344986,
        0.1831699645008743,
        0.00039772156760870626,
        -1.6763828736852209,
        0.00011525622600187983,
        1.8702749392842821,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.00043385578411934294,
          0.2226882123749894,
          0.00013163069572917548,
          -1.7051909194357242,
          -0.00197018274254605,
          1.8530427019958804,
          -0.7853009698993264
        ],
        "ee_pos": [
          0.611361336914154,
          -2.0478782307201376e-06,
          0.35297739527467265
        ],
        "mug_pos": [
          0.5000219465067662,
          3.0959370884570905e-06,
          0.2890711765978987
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0004221920211945396,
        0.1745292760500606,
        0.0004083752030371257,
        -1.684044681189737,
        0.00011790936104243544,
        1.8605967236253629,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.00044169155006643465,
          0.2145325296018996,
          0.0001490463283782246,
          -1.7121906014767496,
          -0.0019766025163743864,
          1.8433415997490412,
          -0.7853007646335398
        ],
        "ee_pos": [
          0.6072447328985021,
          8.078656953228863e-06,
          0.35403140805377015
        ],
        "mug_pos": [
          0.4999892896599701,
          2.546437621363083e-05,
          0.2890719954317187
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0004066754370651564,
        0.16636637336669902,
        0.00039393729785505157,
        -1.6912669729971685,
        0.00011330535272963954,
        1.8514398421096685,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.0004367460889309959,
          0.2062625495682378,
          0.0001537845825393326,
          -1.7192741516325616,
          -0.0019734822272494063,
          1.8339496311899848,
          -0.7852991067211583
        ],
        "ee_pos": [
          0.60313481798888,
          4.273268843776049e-06,
          0.35513088002318216
        ],
        "mug_pos": [
          0.5000242967983398,
          1.067924992229697e-05,
          0.289075076938996
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0003965138323268107,
        0.15835901661452215,
        0.000384612854805791,
        -1.69832857686021,
        0.0001101771708862556,
        1.842450529661555,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.0004266046535316151,
          0.19796360643722566,
          0.00015169381335321656,
          -1.726341821559794,
          -0.00197420604920485,
          1.8251495730918148,
          -0.7852983496257702
        ],
        "ee_pos": [
          0.5991445760031571,
          -7.446049396490597e-06,
          0.35628992930490877
        ],
        "mug_pos": [
          0.5000286399081303,
          6.813286506732439e-06,
          0.289071514945652
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0004048553546526557,
        0.15031246503472007,
        0.0003932312771197752,
        -1.7054191724402932,
        0.00011237424267341135,
        1.8333970123724923,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.00041819149231852524,
          0.18960876670852014,
          0.0001502662190548429,
          -1.7333950495020227,
          -0.001980604342752479,
          1.816286677226627,
          -0.7852987479634793
        ],
        "ee_pos": [
          0.5951384966669728,
          -1.9284547647662056e-05,
          0.3574792274442256
        ],
        "mug_pos": [
          0.49998982201256315,
          2.4954907717819287e-05,
          0.2890740492447676
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.00042210938910932315,
        0.14217382306945872,
        0.00041053914997915233,
        -1.7125868044562442,
        0.00011721109522292179,
        1.82421795428171,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.0004128883420565366,
          0.181185138693006,
          0.00015367139834506234,
          -1.740473796844546,
          -0.0019804078531635246,
          1.8070691251948123,
          -0.7852989500377096
        ],
        "ee_pos": [
          0.5910333835736252,
          -2.54409466636345e-05,
          0.35866805524223017
        ],
        "mug_pos": [
          0.5000253731229197,
          8.41956830786311e-06,
          0.28907460039349503
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0004204692764974739,
        0.13399146638623446,
        0.0004094329502805202,
        -1.7197763981140497,
        0.00011646076125925688,
        1.8149765633525639,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.0004106670017296474,
          0.17273123435910268,
          0.00016157480137779943,
          -1.7475795994481096,
          -0.0019743164259051105,
          1.7977550682487125,
          -0.785298541081054
        ],
        "ee_pos": [
          0.5868798249513552,
          -2.6036101976007722e-05,
          0.35984878996309533
        ],
        "mug_pos": [
          0.5000252750611979,
          6.019204871729163e-06,
          0.2890710297324701
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0004196329385584955,
        0.1258306703196643,
        0.0004090796026299258,
        -1.7269199532961383,
        0.0001159401204811072,
        1.8057543429449627,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.00041086437793943457,
          0.16429588418059018,
          0.000172939355553045,
          -1.754671589294336,
          -0.001972867488767178,
          1.7885998221033377,
          -0.7852982463472775
        ],
        "ee_pos": [
          0.582747939479729,
          -2.383149519288123e-05,
          0.3610267495499627
        ],
        "mug_pos": [
          0.499986743104971,
          2.2418182710763885e-05,
          0.2890727141478841
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.00041648170550500957,
        0.11772617301262041,
        0.00040642683435650727,
        -1.7339849179557252,
        0.00011469985225519827,
        1.796592371034074,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.0004125872550840994,
          0.15590334756109886,
          0.00018604989684226853,
          -1.76171204550702,
          -0.0019775355115011677,
          1.7795945994451825,
          -0.7852983635398858
        ],
        "ee_pos": [
          0.5786507441193499,
          -2.0838219635432067e-05,
          0.3622040207182171
        ],
        "mug_pos": [
          0.5000224930842135,
          5.152482851856804e-06,
          0.2890733602700998
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0004160600851215564,
        0.1096781674311611,
        0.0004064136336531615,
        -1.7409759490494268,
        0.00011429077761865873,
        1.7874875916144437,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.00041512695742271786,
          0.14756522603104807,
          0.00019984917838359567,
          -1.7686803603497652,
          -0.001980852221063155,
          1.7706492292145142,
          -0.7852984659374641
        ],
        "ee_pos": [
          0.5745749008806045,
          -1.6835515052343966e-05,
          0.3633776193750377
        ],
        "mug_pos": [
          0.50002276239403,
          3.524189807350393e-06,
          0.28907114379635274
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.00041769408771400487,
        0.10170553561772902,
        0.00040838874342598145,
        -1.7478758161335919,
        0.00011452218749506865,
        1.7784624509848264,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.000422103317588558,
          0.13965658859546207,
          0.00021828002726311777,
          -1.7751629209294073,
          -0.0019746748996039244,
          1.7626964442058513,
          -0.7852972941705675
        ],
        "ee_pos": [
          0.5708184824414182,
          -6.101654554194105e-06,
          0.36456806927529334
        ],
        "mug_pos": [
          0.4999891546356198,
          2.7197455653400282e-05,
          0.2890721060879465
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.00043067593480254976,
        0.09512141195274174,
        0.000421445925888701,
        -1.7535532310491906,
        0.0001184245121209904,
        1.7710061229327667,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.00043263663507922605,
          0.13166897008228406,
          0.00023696347758674148,
          -1.7818075769890864,
          -0.0019646455710089577,
          1.7544030565418824,
          -0.7852985663739762
        ],
        "ee_pos": [
          0.5669893334319127,
          7.925067495852131e-06,
          0.36570845020728293
        ],
        "mug_pos": [
          0.5000234268216286,
          8.782711332972228e-06,
          0.28907438911140243
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0004642405545230531,
        0.08655059440262022,
        0.00045487002179254684,
        -1.7609112853828255,
        0.00012893231736596798,
        1.7612987593960563,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.0004513535003354098,
          0.12352254102707196,
          0.0002630194145152298,
          -1.7885857482742822,
          -0.001963879334990065,
          1.7448605307435672,
          -0.7852988206788106
        ],
        "ee_pos": [
          0.5628150957702789,
          2.72530248681394e-05,
          0.3668023665603301
        ],
        "mug_pos": [
          0.500026876836383,
          5.0685788017947224e-06,
          0.28907108390448216
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0005106101149767355,
        0.0786265320600497,
        0.0005009457837920469,
        -1.767687413492242,
        0.00014377295523086795,
        1.7523177588163805,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.0004817048402708943,
          0.11578616268580186,
          0.00030384210126270934,
          -1.7949324055703748,
          -0.0019400655254233957,
          1.736602110191786,
          -0.7852959993365759
        ],
        "ee_pos": [
          0.558989464800077,
          6.552613745177066e-05,
          0.3679132863140681
        ],
        "mug_pos": [
          0.4999870815620633,
          2.506279811877329e-05,
          0.28907205432359556
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0005434593995293086,
        0.07228525782665551,
        0.0005336590698643328,
        -1.7730864665001769,
        0.00015434540700321635,
        1.7451304205526126,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.0005202780798616885,
          0.10807333084418289,
          0.0003516610568021997,
          -1.801329784312222,
          -0.0019188915247896003,
          1.728888666232152,
          -0.7852970637225298
        ],
        "ee_pos": [
          0.5553185878699075,
          0.00011272087722869479,
          0.36901139111885284
        ],
        "mug_pos": [
          0.5000217925462669,
          7.34828739318652e-06,
          0.2890730749269448
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0005959367246689026,
        0.0641382925503504,
        0.0005858940448528585,
        -1.779994959530757,
        0.00017149021897231985,
        1.7358937812779216,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.0005684214925595756,
          0.10061890109666499,
          0.00041050048746011714,
          -1.8073567812999318,
          -0.0019197935734779838,
          1.7210624283754896,
          -0.7852979108361428
        ],
        "ee_pos": [
          0.5516883948840454,
          0.0001656350581746423,
          0.3701131483550126
        ],
        "mug_pos": [
          0.5000274117637167,
          7.130793868395832e-06,
          0.28907209163440606
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0006465358727533859,
        0.05787708261254855,
        0.0006362517170257032,
        -1.7852907603113601,
        0.00018827318497172512,
        1.72878710035434,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.0006205772635805567,
          0.09345983068221937,
          0.00047434004317693443,
          -1.8131183920998,
          -0.0018951873860874047,
          1.7138399758737906,
          -0.7852974921842842
        ],
        "ee_pos": [
          0.5482736192175158,
          0.00022858101782479258,
          0.3711912927062537
        ],
        "mug_pos": [
          0.49999163414657943,
          2.4810449310491283e-05,
          0.2890719947258726
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0006881379292872653,
        0.051257250341830665,
        0.0006777520435509817,
        -1.7908711307559082,
        0.00020206970753999237,
        1.72127032348626,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.0006593522399661725,
          0.08609748678400225,
          0.0005283992812996975,
          -1.8191556973086775,
          -0.001889495738118445,
          1.7056579538431147,
          -0.7852996183587659
        ],
        "ee_pos": [
          0.5445963615387772,
          0.00027623957649525963,
          0.3722095726148167
        ],
        "mug_pos": [
          0.5000211322362624,
          9.314858435715847e-06,
          0.28907344959968767
        ],
        "gripper_state": 1.0
      },
      "action": [
        0.0006615588419906872,
        0.043292604989502916,
        0.0006516551559957677,
        -1.7975540946066226,
        0.00019219874835996553,
        1.712225897376757,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.0005713541586156996,
          0.07889513037543723,
          0.0005049642443763663,
          -1.8249936198647112,
          -0.0020342636446086613,
          1.6974264126637306,
          -0.7853061481198811
        ],
        "ee_pos": [
          0.5409055648118708,
          0.00019528860336636013,
          0.3732160821229675
        ],
        "mug_pos": [
          0.5000269647217023,
          5.61442584139154e-06,
          0.2890708312368593
        ],
        "gripper_state": 1.0
      },
      "action": [
        -6.689133495502012e-05,
        0.03719934656978467,
        -7.204163604287837e-05,
        -1.802702184283605,
        -6.431212539114188e-05,
        1.7052624112045072,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          0.00024926402618215135,
          0.07197002303434585,
          0.0002566699593527361,
          -1.8306165143239985,
          -0.002346654908618539,
          1.6904236575266727,
          -0.7853131806694118
        ],
        "ee_pos": [
          0.5375550431900605,
          -0.00015649360580881843,
          0.3742139033112316
        ],
        "mug_pos": [
          0.49998892529201316,
          2.605815011500279e-05,
          0.28907223876323
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0009938837721890593,
        0.03077125016920368,
        -0.0009941466284614155,
        -1.8081322470373145,
        -0.0003932354935453948,
        1.6979004953527121,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.0002502598515381905,
          0.06524235908750906,
          -0.00023756337480975587,
          -1.8360411975183233,
          -0.0026390086217106907,
          1.6837054637377076,
          -0.7853085712353458
        ],
        "ee_pos": [
          0.5343352511856201,
          -0.0007370249373896661,
          0.3751965729907298
        ],
        "mug_pos": [
          0.5000240846729248,
          6.517582401569442e-06,
          0.28907321806958797
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0013805912222858972,
        0.024472647762366184,
        -0.0013793942950271112,
        -1.8133977716567589,
        -0.0005308800648742206,
        1.6907114981941864,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.0008128582312376047,
          0.058671272574498716,
          -0.0008411401682849266,
          -1.8413014679458561,
          -0.0027789121517916276,
          1.676760677529136,
          -0.7852973223925487
        ],
        "ee_pos": [
          0.5311101608302345,
          -0.0013823901322947097,
          0.3761509481302782
        ],
        "mug_pos": [
          0.500027172072999,
          5.29658836610598e-06,
          0.289070974413754
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0016350267511081782,
        0.018340908102684357,
        -0.0016331933953748288,
        -1.8185000115426528,
        -0.0006216371166319755,
        1.68371481202532,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.0013845415271995054,
          0.05223717614506099,
          -0.001445135192027512,
          -1.8464588186898414,
          -0.0028962756353106043,
          1.6697846476274418,
          -0.7852938422912621
        ],
        "ee_pos": [
          0.527901378056405,
          -0.0020217752783766555,
          0.37706968034533006
        ],
        "mug_pos": [
          0.4999915370887213,
          2.587167063281256e-05,
          0.2890715394623084
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0021218291205614206,
        0.012307384666189643,
        -0.002118995462231504,
        -1.8235312926833354,
        -0.000798106023899107,
        1.676806759336154,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.001859203882501221,
          0.046004997537930685,
          -0.001976465838991014,
          -1.8514486474906906,
          -0.0029924099227854528,
          1.663181799633035,
          -0.7852902706496484
        ],
        "ee_pos": [
          0.5248154190613284,
          -0.00257132169947279,
          0.3779617534982831
        ],
        "mug_pos": [
          0.5000247130220786,
          1.0046934292252115e-05,
          0.28907327966149243
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0021005266705061564,
        0.00661197545809017,
        -0.0020978695064540094,
        -1.8282316513714134,
        -0.000788124223986489,
        1.6703082227817798,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.002151992648615841,
          0.040022073386692406,
          -0.002322622117827148,
          -1.8562101594177982,
          -0.00297076408910569,
          1.6570969175816912,
          -0.785284520875309
        ],
        "ee_pos": [
          0.5219120191261887,
          -0.002906669241739574,
          0.3788341522038518
        ],
        "mug_pos": [
          0.5000253561804894,
          9.656410202836442e-06,
          0.2890714426951605
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0019364351501395135,
        0.001183215479483757,
        -0.0019339603989052077,
        -1.8326943796268556,
        -0.0007249055230055254,
        1.6641142179458395,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.0022390005891511546,
          0.034293873405560005,
          -0.002434474966317024,
          -1.8607270213444644,
          -0.0028636567121753786,
          1.6513476580920976,
          -0.7852839176212966
        ],
        "ee_pos": [
          0.5191594519355428,
          -0.002993348072628502,
          0.37968484888633536
        ],
        "mug_pos": [
          0.4999924712665559,
          2.4946944074523172e-05,
          0.28907227155714205
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.001744571031735908,
        -0.0039902243972892845,
        -0.0017420815769595228,
        -1.8369354559653064,
        -0.0006509267628598589,
        1.6582086031192826,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.002151437948733792,
          0.029150454892875877,
          -0.0023339054544973094,
          -1.8646585125778283,
          -0.002750267510992343,
          1.6466818896322157,
          -0.7852877032716981
        ],
        "ee_pos": [
          0.5168022192242719,
          -0.0028753271281626034,
          0.38051479729084803
        ],
        "mug_pos": [
          0.5000199520525148,
          7.066824091149223e-06,
          0.2890733287860695
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0015908229570002102,
        -0.007685440211404391,
        -0.0015881903937644474,
        -1.8399624466871671,
        -0.0005914726558254263,
        1.6539852205028003,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.001943537460953151,
          0.02413597204686074,
          -0.002092733623836935,
          -1.8685835509966828,
          -0.0026449253681491915,
          1.6418710619217458,
          -0.7852934579429102
        ],
        "ee_pos": [
          0.514473057596936,
          -0.002621355699982513,
          0.3812728429399105
        ],
        "mug_pos": [
          0.500026143348333,
          7.56899266158806e-07,
          0.2890701263900713
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0013795129553101455,
        -0.013056118002428414,
        -0.0013765096445570852,
        -1.8443584580278585,
        -0.0005092622673350438,
        1.6478390968039551,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.001676894615811824,
          0.01943859940309656,
          -0.001776737654986893,
          -1.872212011462809,
          -0.002559292179897348,
          1.6368235431233549,
          -0.7852964772850838
        ],
        "ee_pos": [
          0.512142323766737,
          -0.0023008092434349684,
          0.3819803664720025
        ],
        "mug_pos": [
          0.4999830356989501,
          1.5063313784397218e-05,
          0.28907282477592977
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0012386653905644062,
        -0.016622966812840437,
        -0.0012353046129766037,
        -1.847277584839291,
        -0.0004541761234470618,
        1.6437507056080027,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.0014041218417185381,
          0.015133204996181381,
          -0.0014549676423590856,
          -1.875574826763508,
          -0.0024711501824605238,
          1.632687951493306,
          -0.7852978307804118
        ],
        "ee_pos": [
          0.5101043816812856,
          -0.001973376355336301,
          0.3826265753259568
        ],
        "mug_pos": [
          0.5000249390778617,
          7.171925886616857e-06,
          0.2890735536862112
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0010919751630580493,
        -0.020363000183518366,
        -0.0010881567326418956,
        -1.8503386275442568,
        -0.0003965645727588161,
        1.6394578870739314,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.00115546882260541,
          0.011081776299238174,
          -0.0011678708230835504,
          -1.8787718981664359,
          -0.0024044505877640968,
          1.628816614987151,
          -0.7852996225958133
        ],
        "ee_pos": [
          0.5081933354918363,
          -0.0016808227342550261,
          0.3832179655605099
        ],
        "mug_pos": [
          0.5000249919662373,
          5.451455480140305e-06,
          0.28907132865219626
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.000937930803574209,
        -0.02409431897375583,
        -0.000933550096531879,
        -1.8533914113276826,
        -0.00033585081532031794,
        1.6351699123255619,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.0009248844711684616,
          0.007201030606189773,
          -0.0009205876730488785,
          -1.8818489988230342,
          -0.00233082946345636,
          1.6247359520616222,
          -0.7852989675682857
        ],
        "ee_pos": [
          0.5062753868636937,
          -0.0014198897123345966,
          0.38376347125390725
        ],
        "mug_pos": [
          0.4999901391655923,
          1.9973803008608457e-05,
          0.28907299807622355
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.0006745592723958497,
        -0.027673563710695015,
        -0.0006691559753966575,
        -1.8563091097908782,
        -0.0002321141175933283,
        1.6310589131558064,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.0006978737039304942,
          0.0034855538125155277,
          -0.0006890491778800306,
          -1.8848112223451141,
          -0.0022266019567661354,
          1.6207130875946563,
          -0.7852966900335343
        ],
        "ee_pos": [
          0.5044011020930752,
          -0.0011653756770579082,
          0.38427261170058047
        ],
        "mug_pos": [
          0.5000239984259222,
          7.674438001425327e-06,
          0.28907373998377395
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.00040058748163960243,
        -0.031048915381524173,
        -0.0003940129380378024,
        -1.859056377828437,
        -0.0001238286948810117,
        1.6271800833612988,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.0004809343017602716,
          -4.3446982633584734e-05,
          -0.00046302643350434327,
          -1.8876262290722419,
          -0.0021354616071762845,
          1.6170574663748605,
          -0.7852968542574266
        ],
        "ee_pos": [
          0.5026525027580944,
          -0.0009214661028313702,
          0.38475802998473835
        ],
        "mug_pos": [
          0.5000263198066138,
          5.303022724657879e-06,
          0.28907128795766573
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.00023364999706747795,
        -0.034209048652194034,
        -0.00022628565542027437,
        -1.8616341170181614,
        -5.753619107188272e-05,
        1.6235398468606672,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.0002937850078896513,
          -0.00302066746817998,
          -0.0002584088777053965,
          -1.8898872950053835,
          -0.0020965415686724147,
          1.6146951776537373,
          -0.7852982539258491
        ],
        "ee_pos": [
          0.5013385074416806,
          -0.000713600700574864,
          0.38523513659821096
        ],
        "mug_pos": [
          0.49998888985830847,
          2.6233810891512758e-05,
          0.2890722364315258
        ],
        "gripper_state": 1.0
      },
      "action": [
        -0.00016741213979155315,
        -0.03585189684429429,
        -0.00015971396243844824,
        -1.862977313543982,
        -3.11496400002489e-05,
        1.6216434021922415,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -0.00014316126696187686,
          -0.005835180183509066,
          -9.876845633939712e-05,
          -1.8920747241491118,
          -0.002071357551557681,
          1.6124063878951513,
          -0.7852999547388485
        ],
        "ee_pos": [
          0.5001214179570204,
          -0.0005505329140328043,
          0.38566341174892566
        ],
        "mug_pos": [
          0.5000267283077916,
          3.8311873774575055e-06,
          0.28907218890187303
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5466633701718e-05,
        -0.038988332063644826,
        -3.7125076103255846e-05,
        -1.8655396794462007,
        1.7551384012418424e-05,
        1.6180209907263399,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -3.3823123021198064e-05,
          -0.008334551260326454,
          1.9810071363158233e-05,
          -1.8939517014441338,
          -0.0020572637826593355,
          1.6098694011605725,
          -0.7852995142188904
        ],
        "ee_pos": [
          0.4989129081157239,
          -0.00043143999061880296,
          0.3860555677860137
        ],
        "mug_pos": [
          0.5000253607535964,
          6.394111062681502e-06,
          0.2890705031786443
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.9663512207589986e-05,
        -0.04047799259140307,
        -1.1178887262041535e-05,
        -1.8667586728910452,
        2.7876531631995796e-05,
        1.6162975067684453,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          3.237885296474109e-05,
          -0.01042452428826915,
          9.142278915282378e-05,
          -1.895534265841691,
          -0.002042363000703085,
          1.6081783644859475,
          -0.7852984303145092
        ],
        "ee_pos": [
          0.49799134204155343,
          -0.00035719320815221173,
          0.3863904065645082
        ],
        "mug_pos": [
          0.49999253249623327,
          2.488662981149259e-05,
          0.28907183641489037
        ],
        "gripper_state": 1.0
      },
      "action": [
        -8.205147670246228e-06,
        -0.04198507463353113,
        3.445334171029554e-07,
        -1.867992893281055,
        3.2457050822078816e-05,
        1.614552167834196,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          6.144594932288665e-05,
          -0.01220940644836196,
          0.00012049000988473723,
          -1.8969116271870183,
          -0.0020490266304115006,
          1.606876850322949,
          -0.7852986207691137
        ],
        "ee_pos": [
          0.49723875731851813,
          -0.0003274626743968637,
          0.3866688488340581
        ],
        "mug_pos": [
          0.5000264911452356,
          7.262244906324793e-06,
          0.28907296806626087
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.410874354943281e-06,
        -0.04342604300115831,
        4.160306101408941e-06,
        -1.8691725057003636,
        3.3960692650273665e-05,
        1.6128827876307226,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          6.323034082852365e-05,
          -0.013782190132693172,
          0.00011767991995088695,
          -1.898147810662566,
          -0.0020667214496882837,
          1.6054498821569585,
          -0.7852984869805628
        ],
        "ee_pos": [
          0.4965105058125966,
          -0.0003305016562581033,
          0.3868964946442267
        ],
        "mug_pos": [
          0.5000248550920963,
          7.824284481260658e-06,
          0.2890703776721496
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          5.1558391832367296e-05,
          -0.014867037131109883,
          9.93663241489516e-05,
          -1.8989474904572372,
          -0.0020744229030988785,
          1.6048352234631722,
          -0.7852965284866232
        ],
        "ee_pos": [
          0.4960754786875148,
          -0.0003468845248759452,
          0.38708624094662714
        ],
        "mug_pos": [
          0.4999958835291601,
          2.1723724890159263e-05,
          0.2890714027835844
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          3.512932155769026e-05,
          -0.015449055146529384,
          7.625774296270595e-05,
          -1.899344951976529,
          -0.002070699476756474,
          1.6053024759007388,
          -0.7852959621212734
        ],
        "ee_pos": [
          0.4960286899231968,
          -0.00036555403208103794,
          0.3872250070684965
        ],
        "mug_pos": [
          0.5000271414503169,
          3.973041934352033e-06,
          0.2890717224473721
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          1.9099734216395802e-05,
          -0.015716293376892024,
          5.44969618994124e-05,
          -1.8995077260925788,
          -0.002075802468846372,
          1.6057483147141394,
          -0.7852971575027774
        ],
        "ee_pos": [
          0.4960734975164272,
          -0.0003853075581518624,
          0.38730568561140083
        ],
        "mug_pos": [
          0.5000271252816358,
          3.4027953575717843e-06,
          0.2890695965562303
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          5.8163536730660295e-06,
          -0.01582136781132969,
          3.6757142300101966e-05,
          -1.899595077085505,
          -0.0020802880139711016,
          1.605711876199265,
          -0.7852973720666598
        ],
        "ee_pos": [
          0.4960414108012937,
          -0.00040197338608692157,
          0.3873239699492547
        ],
        "mug_pos": [
          0.49999122991189304,
          2.226932529545422e-05,
          0.2890716762348491
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -3.5690558859498553e-06,
          -0.015838627445354806,
          2.4358929533665452e-05,
          -1.8996860739658739,
          -0.002077709233274057,
          1.6055185882883918,
          -0.7852965551841874
        ],
        "ee_pos": [
          0.4959728225849262,
          -0.00041265958715589477,
          0.3872914358289069
        ],
        "mug_pos": [
          0.5000231183484976,
          4.8388924635306685e-06,
          0.2890723494337644
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -8.733685202865338e-06,
          -0.015804767821565507,
          1.7657146658384873e-05,
          -1.899778383251379,
          -0.0020773525729127656,
          1.6054640437604584,
          -0.7852962348736551
        ],
        "ee_pos": [
          0.4959355856569591,
          -0.00041870561623582605,
          0.38723504787267726
        ],
        "mug_pos": [
          0.5000268304121096,
          2.8433281887483296e-06,
          0.28907039414358354
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.046612616670536e-05,
          -0.015751983209445108,
          1.5548717399039643e-05,
          -1.8998489648873034,
          -0.0020823411103456134,
          1.6055149704112979,
          -0.7852965483823829
        ],
        "ee_pos": [
          0.4959302385785562,
          -0.00042178883657292886,
          0.38718009701954315
        ],
        "mug_pos": [
          0.49999213550850324,
          2.558174311074734e-05,
          0.28907127469828753
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0122901994650451e-05,
          -0.01570649663744258,
          1.6173553540475015e-05,
          -1.899892169839527,
          -0.0020855988894281194,
          1.6055469412680694,
          -0.785296779492835
        ],
        "ee_pos": [
          0.4959288096816779,
          -0.0004221321959139837,
          0.3871393158291708
        ],
        "mug_pos": [
          0.5000260494831221,
          6.2548745481083334e-06,
          0.28907272050599875
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -8.814689809622028e-06,
          -0.01567945939090991,
          1.8028003491756903e-05,
          -1.8999155580401663,
          -0.002083687376827015,
          1.6055358919193325,
          -0.7852967558784079
        ],
        "ee_pos": [
          0.49592182163116044,
          -0.00042026768631944615,
          0.38711485319111694
        ],
        "mug_pos": [
          0.5000261568613187,
          5.289214425711832e-06,
          0.2890704885636417
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -7.203304243833685e-06,
          -0.01566928353299718,
          2.0238505950600994e-05,
          -1.8999260243792406,
          -0.002080452958119719,
          1.6055224911040844,
          -0.7852967272398015
        ],
        "ee_pos": [
          0.49591629964461,
          -0.00041770655527252295,
          0.3871043029398333
        ],
        "mug_pos": [
          0.49999384076155345,
          2.176409468622084e-05,
          0.28907275018680956
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -5.708818428199092e-06,
          -0.0156701096614533,
          2.2261896373027628e-05,
          -1.8999269451445235,
          -0.002079345550367321,
          1.605526664718444,
          -0.7852967790497729
        ],
        "ee_pos": [
          0.4959166292859292,
          -0.00041568357845884243,
          0.38710394247747804
        ],
        "mug_pos": [
          0.5000235587511368,
          7.093889824566975e-06,
          0.28907349466257026
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.6083016867471576e-06,
          -0.015676850201947135,
          2.3739297998337257e-05,
          -1.899921553073522,
          -0.0020796308704792463,
          1.6055368681714677,
          -0.7852968237013296
        ],
        "ee_pos": [
          0.49591992618938235,
          -0.0004144205433327865,
          0.3871095866649761
        ],
        "mug_pos": [
          0.5000258191260403,
          4.111725622770493e-06,
          0.28907071618192254
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -3.990404931459379e-06,
          -0.015685772974198972,
          2.4559203549674067e-05,
          -1.899913925372263,
          -0.002079537168535807,
          1.6055410050731869,
          -0.7852968222046826
        ],
        "ee_pos": [
          0.49592247468714873,
          -0.0004136619878171467,
          0.38711729676901485
        ],
        "mug_pos": [
          0.4999889131708622,
          2.1708582726084684e-05,
          0.2890716683994677
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -3.772714526326538e-06,
          -0.015694069519792592,
          2.4838744834467156e-05,
          -1.899907282024737,
          -0.002078982886174113,
          1.6055398357858717,
          -0.7852968065483624
        ],
        "ee_pos": [
          0.495923575843453,
          -0.00041327915488058526,
          0.38712426348434215
        ],
        "mug_pos": [
          0.500022759545037,
          7.476020949474926e-06,
          0.2890727545846844
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -3.809541358287759e-06,
          -0.0157000297837092,
          2.477862727814183e-05,
          -1.8999029189593426,
          -0.0020787491916822404,
          1.6055385167378369,
          -0.7852967987693037
        ],
        "ee_pos": [
          0.49592411806743486,
          -0.0004132592624089183,
          0.3871291530820214
        ],
        "mug_pos": [
          0.5000261343270396,
          4.319275331565378e-06,
          0.28907089493451193
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -3.977112247149189e-06,
          -0.01570322550708882,
          2.4547037180859825e-05,
          -1.8999007414019282,
          -0.0020789967381397707,
          1.6055385680178644,
          -0.7852967941386393
        ],
        "ee_pos": [
          0.49592453975621686,
          -0.00041349771987348783,
          0.38713178150571426
        ],
        "mug_pos": [
          0.4999894091391778,
          1.999340469849433e-05,
          0.28907310287031374
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.18939871188364e-06,
          -0.015704158383603607,
          2.4258975751547114e-05,
          -1.899900193307311,
          -0.0020793146070276023,
          1.6055387063069926,
          -0.785296787707261
        ],
        "ee_pos": [
          0.4959246870670783,
          -0.0004138114432538819,
          0.38713257202155055
        ],
        "mug_pos": [
          0.5000262904914067,
          9.452889373766597e-06,
          0.2890741182622738
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.386715876942173e-06,
          -0.015703662361913166,
          2.3993390255698754e-05,
          -1.8999006949164572,
          -0.002079448087652978,
          1.6055382067631447,
          -0.7852967823185981
        ],
        "ee_pos": [
          0.4959244871595388,
          -0.0004140735491430273,
          0.3871321562734419
        ],
        "mug_pos": [
          0.5000272031792484,
          7.055353710322936e-06,
          0.2890713619879312
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.533257231119822e-06,
          -0.015702498210996275,
          2.379743814362311e-05,
          -1.899901708100676,
          -0.0020794726058686363,
          1.6055375382361485,
          -0.7852967802632618
        ],
        "ee_pos": [
          0.4959241354274173,
          -0.0004142535411237057,
          0.38713115508960805
        ],
        "mug_pos": [
          0.49999211875149885,
          2.3529136065796754e-05,
          0.28907329200947973
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.618812051842757e-06,
          -0.01570121910923248,
          2.368413440136461e-05,
          -1.899902779502346,
          -0.00207950509871169,
          1.605537212678023,
          -0.7852967803534633
        ],
        "ee_pos": [
          0.4959238427128867,
          -0.00041436293716771695,
          0.3871300574969062
        ],
        "mug_pos": [
          0.5000259085991481,
          6.100654201736917e-06,
          0.28907353613675957
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.652328107892304e-06,
          -0.0157001641786702,
          2.3640850993641292e-05,
          -1.8999036213923246,
          -0.002079543362294093,
          1.6055372093040086,
          -0.7852967810696289
        ],
        "ee_pos": [
          0.495923670701193,
          -0.0004144123351072754,
          0.38712916368569
        ],
        "mug_pos": [
          0.5000265678559711,
          2.5760086377645932e-06,
          0.2890709123650169
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.650289847305194e-06,
          -0.01569948078191241,
          2.364501761440854e-05,
          -1.8999041341363871,
          -0.0020795452959855446,
          1.6055372897574407,
          -0.785296782054665
        ],
        "ee_pos": [
          0.49592358660990027,
          -0.00041441219007364515,
          0.3871285917453241
        ],
        "mug_pos": [
          0.49999323919189365,
          2.465551602456005e-05,
          0.28907127623150075
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.6285667315402916e-06,
          -0.015699165921852425,
          2.3675157363403025e-05,
          -1.8999043492859713,
          -0.0020795142235416884,
          1.6055373470848935,
          -0.7852967832189528
        ],
        "ee_pos": [
          0.49592355701473856,
          -0.0004143812826941717,
          0.38712832998475893
        ],
        "mug_pos": [
          0.5000252978851438,
          6.724682774089464e-06,
          0.2890725377524445
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.599710718998423e-06,
          -0.01569912868410792,
          2.3714445233505568e-05,
          -1.899904353433476,
          -0.002079481783643782,
          1.6055374014933672,
          -0.7852967842267011
        ],
        "ee_pos": [
          0.49592356795625137,
          -0.0004143408600270907,
          0.38712829999086035
        ],
        "mug_pos": [
          0.5000257354208938,
          4.205277030735494e-06,
          0.2890701519820585
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.572565838166233e-06,
          -0.01569925272263872,
          2.3751105942024884e-05,
          -1.8999042391698449,
          -0.0020794638180727325,
          1.6055374690684796,
          -0.7852967848327594
        ],
        "ee_pos": [
          0.4959236054035609,
          -0.000414304980824396,
          0.3871284057130982
        ],
        "mug_pos": [
          0.49999186146512753,
          2.1533232317251068e-05,
          0.2890720712181656
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.552131017172374e-06,
          -0.015699437219936246,
          2.377853120236999e-05,
          -1.8999040838778134,
          -0.0020794557563801443,
          1.6055375272070982,
          -0.7852967850768249
        ],
        "ee_pos": [
          0.495923649479333,
          -0.000414278911616477,
          0.3871285628305442
        ],
        "mug_pos": [
          0.5000230259945254,
          5.962677807620921e-06,
          0.2890729749141855
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.539882091259595e-06,
          -0.015699613172445965,
          2.3794835404809165e-05,
          -1.8999039412985221,
          -0.0020794511057408327,
          1.6055375553693627,
          -0.7852967851037906
        ],
        "ee_pos": [
          0.4959236846027365,
          -0.0004142631672561422,
          0.3871287123107084
        ],
        "mug_pos": [
          0.5000260609504655,
          1.2903907577444494e-06,
          0.289070395710609
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.53485859250775e-06,
          -0.015699744064286935,
          2.38013911936959e-05,
          -1.8999038392915109,
          -0.002079449384243614,
          1.60553755746537,
          -0.7852967850067374
        ],
        "ee_pos": [
          0.49592370568091115,
          -0.00041425656004911613,
          0.387128822866499
        ],
        "mug_pos": [
          0.4999904511943656,
          2.3004861714369628e-05,
          0.2890718130760679
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.534992273702757e-06,
          -0.015699819575464638,
          2.3801034382561466e-05,
          -1.8999037837284165,
          -0.0020794514085757778,
          1.6055375482613838,
          -0.7852967848403607
        ],
        "ee_pos": [
          0.4959237147801581,
          -0.00041425687806241916,
          0.3871288861557467
        ],
        "mug_pos": [
          0.5000243343691619,
          6.968229188272309e-06,
          0.28907232203954825
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.5380711403280336e-06,
          -0.01569984669423124,
          2.3796764980268984e-05,
          -1.8999037666894674,
          -0.0020794555684304893,
          1.6055375371588834,
          -0.7852967846628545
        ],
        "ee_pos": [
          0.49592371570879223,
          -0.00041426122121069253,
          0.3871289085808997
        ],
        "mug_pos": [
          0.500027253068056,
          5.395524629262925e-06,
          0.28907042825363316
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.5422302599989035e-06,
          -0.015699840640872948,
          2.379109469398406e-05,
          -1.899903774827054,
          -0.002079459406146854,
          1.6055375267644783,
          -0.7852967845235762
        ],
        "ee_pos": [
          0.4959237117779776,
          -0.00041426687404063386,
          0.387128903159069
        ],
        "mug_pos": [
          0.4999920645569517,
          2.5213536737424062e-05,
          0.28907166976110227
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.546152585618168e-06,
          -0.015699817614843934,
          2.378578708791966e-05,
          -1.8999037950722537,
          -0.0020794618409579286,
          1.6055375180235036,
          -0.7852967844403737
        ],
        "ee_pos": [
          0.4959237057885668,
          -0.00041427202203386934,
          0.3871288835180476
        ],
        "mug_pos": [
          0.5000263891645351,
          6.6571718073599715e-06,
          0.28907253435732533
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.549110850636167e-06,
          -0.015699790501071113,
          2.3781808081727182e-05,
          -1.899903817432673,
          -0.002079463086320244,
          1.6055375122050506,
          -0.7852967844052824
        ],
        "ee_pos": [
          0.4959237000026199,
          -0.00041427581663204755,
          0.3871288604977693
        ],
        "mug_pos": [
          0.5000261048627372,
          7.357963801236644e-06,
          0.2890706667725835
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.55088610730026e-06,
          -0.015699767362130794,
          2.37794391149915e-05,
          -1.8999038358210885,
          -0.0020794635645064222,
          1.605537509808692,
          -0.7852967844032124
        ],
        "ee_pos": [
          0.49592369576307327,
          -0.00041427806524518905,
          0.38712884091320904
        ],
        "mug_pos": [
          0.49999362261820013,
          2.515255768584881e-05,
          0.2890719102147985
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.551604132753943e-06,
          -0.015699751763338696,
          2.3778499554134582e-05,
          -1.8999038477323935,
          -0.0020794634839736985,
          1.60553751000817,
          -0.785296784421108
        ],
        "ee_pos": [
          0.49592369340576115,
          -0.00041427894595925886,
          0.3871288277680107
        ],
        "mug_pos": [
          0.5000207124280662,
          1.2008657768725207e-05,
          0.2890739814379922
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.5515597646768224e-06,
          -0.015699743961995766,
          2.3778584568602788e-05,
          -1.8999038532778787,
          -0.0020794630222953758,
          1.6055375114633605,
          -0.7852967844478528
        ],
        "ee_pos": [
          0.49592369261323305,
          -0.0004142788347583728,
          0.38712882124237946
        ],
        "mug_pos": [
          0.500027096031713,
          1.0890964807440445e-05,
          0.2890710791074407
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.551081953012971e-06,
          -0.01569974230377611,
          2.3779247071535364e-05,
          -1.899903854008684,
          -0.0020794624337279032,
          1.6055375131821943,
          -0.7852967844740739
        ],
        "ee_pos": [
          0.4959236928483713,
          -0.00041427816934890954,
          0.3871288199025038
        ],
        "mug_pos": [
          0.49999315143868933,
          2.7545737908503644e-05,
          0.28907261668114326
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.550452349568052e-06,
          -0.015699744444019507,
          2.3780106618728663e-05,
          -1.899903851902107,
          -0.0020794619364145226,
          1.6055375146780753,
          -0.7852967844938817
        ],
        "ee_pos": [
          0.4959236936032498,
          -0.0004142773295013716,
          0.3871288217525145
        ],
        "mug_pos": [
          0.5000239502253099,
          1.2153774806336203e-05,
          0.2890733283372937
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.5498676649693435e-06,
          -0.015699748198601433,
          2.3780899045195315e-05,
          -1.8999038487059732,
          -0.0020794616173717153,
          1.605537515751963,
          -0.7852967845055436
        ],
        "ee_pos": [
          0.49592369448343754,
          -0.00041427657023419893,
          0.387128824947768
        ],
        "mug_pos": [
          0.5000299703492455,
          1.0277246931719826e-05,
          0.2890715440485928
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.5494335396876305e-06,
          -0.01569975198202812,
          2.3781483792030096e-05,
          -1.8999038456401713,
          -0.0020794614628135938,
          1.605537516346118,
          -0.7852967845101255
        ],
        "ee_pos": [
          0.495923695231889,
          -0.00041427601812734467,
          0.3871288281531342
        ],
        "mug_pos": [
          0.49999241467385774,
          1.9829301798643693e-05,
          0.28907235563532285
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.549179490757957e-06,
          -0.015699754908198558,
          2.3781823086367633e-05,
          -1.8999038433541682,
          -0.0020794614284197666,
          1.6055375165181753,
          -0.7852967845097196
        ],
        "ee_pos": [
          0.4959236957304083,
          -0.00041427570237696873,
          0.3871288306238988
        ],
        "mug_pos": [
          0.5000260930067741,
          6.239074123790612e-06,
          0.2890727969781607
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.549084247994949e-06,
          -0.01569975668259097,
          2.3781947342600394e-05,
          -1.899903842030459,
          -0.002079461471445298,
          1.605537516409395,
          -0.7852967845064673
        ],
        "ee_pos": [
          0.49592369597293506,
          -0.0004142755912948491,
          0.3871288321154299
        ],
        "mug_pos": [
          0.5000260600705647,
          7.017040245608199e-06,
          0.2890708852953999
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.5491016437686785e-06,
          -0.015699757406805123,
          2.378192002672731e-05,
          -1.8999038415479748,
          -0.002079461552348346,
          1.605537516175074,
          -0.7852967845022145
        ],
        "ee_pos": [
          0.4959236960181243,
          -0.0004142756233509384,
          0.38712883271797977
        ],
        "mug_pos": [
          0.49999138267976123,
          2.422304899942073e-05,
          0.2890717343342025
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.549180869378493e-06,
          -0.015699757379671945,
          2.3781810305840802e-05,
          -1.8999038416443397,
          -0.002079461636801019,
          1.6055375159293,
          -0.7852967844983091
        ],
        "ee_pos": [
          0.49592369594425273,
          -0.0004142757316594216,
          0.38712883268723947
        ],
        "mug_pos": [
          0.5000275009436854,
          5.059623915015253e-06,
          0.28907326426835667
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.54927912792793e-06,
          -0.015699756941073544,
          2.378167608117354e-05,
          -1.8999038420398635,
          -0.0020794617027601746,
          1.605537515732981,
          -0.7852967844954796
        ],
        "ee_pos": [
          0.4959236958211981,
          -0.0004142758606744089,
          0.3871288323116537
        ],
        "mug_pos": [
          0.5000272969873538,
          5.231537220399225e-06,
          0.2890703753976568
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.549367359077628e-06,
          -0.015699756375130033,
          2.3781556418952022e-05,
          -1.8999038425102568,
          -0.002079461742799255,
          1.6055375156075298,
          -0.7852967844938951
        ],
        "ee_pos": [
          0.4959236956987899,
          -0.00041427597381367334,
          0.38712883183114255
        ],
        "mug_pos": [
          0.4999937900342531,
          2.4483938377237634e-05,
          0.2890721261909884
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.5494307510664034e-06,
          -0.015699755869144384,
          2.378147100980457e-05,
          -1.899903842913712,
          -0.0020794617595152774,
          1.605537515550248,
          -0.7852967844933684
        ],
        "ee_pos": [
          0.49592369560491845,
          -0.0004142760534153832,
          0.38712883140326226
        ],
        "mug_pos": [
          0.5000246648738117,
          5.708019029455467e-06,
          0.28907264414607625
        ],
        "gripper_state": 1.0
      },
      "action": [
        -4.5494235850747705e-06,
        -0.04473579913109608,
        4.019898264688778e-06,
        -1.8702434371771361,
        3.388203637221325e-05,
        1.6113655283221615,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -6.784791671571011e-06,
          -0.01588703609206781,
          2.1103091367754187e-05,
          -1.9002968140414105,
          -0.002063438581889672,
          1.6048252200718411,
          -0.78529764228523
        ],
        "ee_pos": [
          0.4956755827613576,
          -0.00041355747874849175,
          0.3870462326472145
        ],
        "mug_pos": [
          0.5000260761380336,
          2.4983693910946086e-06,
          0.28907040996233613
        ],
        "gripper_state": 1.0
      },
      "action": [
        -6.4907550056030586e-06,
        -0.045045726103608186,
        2.0661901621178446e-06,
        -1.871721702752014,
        3.306942710606859e-05,
        1.6101013792198873,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -7.228133063172467e-06,
          -0.01602463292491043,
          2.0684987250197548e-05,
          -1.9007705899759335,
          -0.002067860116801952,
          1.6039369735665026,
          -0.7852968886785354
        ],
        "ee_pos": [
          0.4953349625487231,
          -0.0004149867985200803,
          0.3868934338473282
        ],
        "mug_pos": [
          0.4999887237770635,
          1.99767732423179e-05,
          0.28907282140251955
        ],
        "gripper_state": 1.0
      },
      "action": [
        -6.4907550056030586e-06,
        -0.045045726103608186,
        2.0661901621178446e-06,
        -1.871721702752014,
        3.306942710606859e-05,
        1.6101013792198873,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -7.535141798391195e-06,
          -0.01605859793443885,
          2.0573499312633667e-05,
          -1.901467117725934,
          -0.0020544832165601983,
          1.6034884853683316,
          -0.7852964200506093
        ],
        "ee_pos": [
          0.49505270121512956,
          -0.0004128042757492326,
          0.38661919109277393
        ],
        "mug_pos": [
          0.5000227482478994,
          3.787944416603784e-06,
          0.28907310444128614
        ],
        "gripper_state": 1.0
      },
      "action": [
        -7.813285481488883e-06,
        -0.04452591142114439,
        7.35761205983761e-07,
        -1.8732972520937223,
        3.250692387730472e-05,
        1.6092266253936034,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -6.545220854421369e-06,
          -0.015850757635757072,
          2.2102672255610208e-05,
          -1.9024134330263263,
          -0.002040076298571173,
          1.6029943483853235,
          -0.7852972478609099
        ],
        "ee_pos": [
          0.494708317119263,
          -0.00040868739436942777,
          0.38612330351360236
        ],
        "mug_pos": [
          0.5000282402046548,
          2.0609976500038116e-06,
          0.2890709027955883
        ],
        "gripper_state": 1.0
      },
      "action": [
        -8.713895139840963e-06,
        -0.04371162078129861,
        -1.6984299414456984e-07,
        -1.874720989693931,
        3.2118569315708834e-05,
        1.6086285499170696,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.215917560253393e-06,
          -0.015291266229788943,
          2.5232182864858887e-05,
          -1.9035415615581845,
          -0.0020384908597486325,
          1.6025226606329137,
          -0.7852974545197485
        ],
        "ee_pos": [
          0.4943339648218292,
          -0.00040565279617761727,
          0.38537911665623464
        ],
        "mug_pos": [
          0.4999885436224852,
          2.1954313468905605e-05,
          0.2890719711683067
        ],
        "gripper_state": 1.0
      },
      "action": [
        -9.326658573577007e-06,
        -0.04261150298027929,
        -7.856192099542894e-07,
        -1.8760604377853074,
        3.184907725301378e-05,
        1.6082525854173129,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.618520626658839e-06,
          -0.014347231033409454,
          2.8400761933261776e-05,
          -1.9047873447366839,
          -0.002044480925207915,
          1.6022876894427833,
          -0.7852970353906781
        ],
        "ee_pos": [
          0.49399686051816727,
          -0.0004039873659724873,
          0.3844013020241243
        ],
        "mug_pos": [
          0.5000236849004898,
          4.416445773475455e-06,
          0.2890727495171309
        ],
        "gripper_state": 1.0
      },
      "action": [
        -9.743156868238624e-06,
        -0.04123637205533667,
        -1.203782779331417e-06,
        -1.877363076845477,
        3.1660558117504366e-05,
        1.6080577489151275,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          7.613707074327283e-07,
          -0.012882186747813888,
          3.069166973972962e-05,
          -1.9062155589557048,
          -0.002025856377040411,
          1.602399423909009,
          -0.7852970187377267
        ],
        "ee_pos": [
          0.49371407835955494,
          -0.00039817844332468025,
          0.38311122365969896
        ],
        "mug_pos": [
          0.5000271063621183,
          3.983949519495205e-06,
          0.28907051489130564
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0218063980621204e-05,
        -0.03845537148279689,
        -1.6799224337120247e-06,
        -1.879535030756836,
        3.1436592049653954e-05,
        1.6080067615861897,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          2.774243971176847e-06,
          -0.010985171560361356,
          3.216395261018079e-05,
          -1.9077446018989241,
          -0.002036397257995581,
          1.6025310007333657,
          -0.7852974645478193
        ],
        "ee_pos": [
          0.4934291127251161,
          -0.0003978614940096832,
          0.38155317849555515
        ],
        "mug_pos": [
          0.49999031939419736,
          2.427768113924448e-05,
          0.2890716513892518
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0347996858078242e-05,
        -0.03690939364347729,
        -1.8098399858137313e-06,
        -1.8805639503329803,
        3.1369684997076914e-05,
        1.608110817771258,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          3.5087138309283917e-06,
          -0.008680751361568297,
          3.122530350797231e-05,
          -1.9093242835468631,
          -0.0020444562475774235,
          1.6028482183474473,
          -0.7852970753009952
        ],
        "ee_pos": [
          0.49318031361669423,
          -0.0003998202177532274,
          0.37978613855586
        ],
        "mug_pos": [
          0.5000277914523592,
          5.387977000214357e-06,
          0.28907225315615404
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0494680162459503e-05,
        -0.03368484287694472,
        -1.9559244840417774e-06,
        -1.882478674949903,
        3.128466301350881e-05,
        1.608499564166301,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          2.2949734001063777e-06,
          -0.006145331562929663,
          2.734343537721838e-05,
          -1.910918483011464,
          -0.002060312014948398,
          1.6032025969510126,
          -0.7852968495692677
        ],
        "ee_pos": [
          0.4929452610675235,
          -0.0004050225319201889,
          0.37788670167477056
        ],
        "mug_pos": [
          0.5000243708119887,
          8.957036702374745e-06,
          0.2890708056098425
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0533982247051398e-05,
        -0.03188289431473119,
        -1.9947372957717385e-06,
        -1.883474567127853,
        3.125577466759693e-05,
        1.6087719369096543,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          6.50380059480407e-08,
          -0.003353455613661238,
          2.1483810612324417e-05,
          -1.912483479160397,
          -0.00205372736862835,
          1.6037818425437036,
          -0.7852964984611744
        ],
        "ee_pos": [
          0.49276459836010555,
          -0.0004080597214162413,
          0.3758898988203078
        ],
        "mug_pos": [
          0.49999191229631046,
          2.239504943358346e-05,
          0.2890719865388902
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0576696540649084e-05,
        -0.028147418974200207,
        -2.0363739515481105e-06,
        -1.885408690566699,
        3.121385136586729e-05,
        1.6094335482551605,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.6712138873592473e-06,
          -0.00020697005383084682,
          1.5932858766022286e-05,
          -1.9140811104993891,
          -0.0020315486787180737,
          1.6046745651726815,
          -0.7852966279845788
        ],
        "ee_pos": [
          0.49265546956750883,
          -0.0004069090134684099,
          0.37371338180270586
        ],
        "mug_pos": [
          0.5000262275505971,
          5.834097296606387e-06,
          0.28907248634624266
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0593938360021408e-05,
        -0.02462192608780436,
        -2.052646416993031e-06,
        -1.8871108628229898,
        3.1185725019881215e-05,
        1.610149611523185,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.5896028134672806e-06,
          0.0032193366175438474,
          1.1144098749331805e-05,
          -1.9157203227804132,
          -0.0020318755432720014,
          1.6055568892504213,
          -0.7852970200021865
        ],
        "ee_pos": [
          0.49253743374182146,
          -0.0004093272622824438,
          0.3713803743913925
        ],
        "mug_pos": [
          0.5000267283729397,
          6.87620529469104e-06,
          0.2890705226148202
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0599300182917553e-05,
        -0.021114502378725718,
        -2.0571368938316064e-06,
        -1.8887364816935037,
        3.116365167331799e-05,
        1.6109125666781319,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -3.622290118920068e-06,
          0.006833390837871856,
          5.8487373378928005e-06,
          -1.9173961940687352,
          -0.0020438915696148903,
          1.6063432528930226,
          -0.7852968305529521
        ],
        "ee_pos": [
          0.4923823029747672,
          -0.00041469531007996493,
          0.36893688632240945
        ],
        "mug_pos": [
          0.4999918723065341,
          2.2785259053472507e-05,
          0.2890718316672125
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0599047971893274e-05,
        -0.017536702366387737,
        -2.056115378867173e-06,
        -1.8903514823794314,
        3.114415479936013e-05,
        1.6117231664516058,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -5.209083484457557e-06,
          0.01057628299085534,
          -4.964058507060548e-07,
          -1.9190900207075892,
          -0.0020470893103797442,
          1.6071646585186892,
          -0.7852963516948057
        ],
        "ee_pos": [
          0.49221964667644647,
          -0.0004191927140046813,
          0.36642316934770225
        ],
        "mug_pos": [
          0.500026546770563,
          5.810532252618006e-06,
          0.2890724773008822
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0596067333821934e-05,
        -0.013850408532154019,
        -2.0524788797799056e-06,
        -1.8919810195830364,
        3.1125712740575236e-05,
        1.6125841378419699,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -6.916567227965114e-06,
          0.0144188246098049,
          -7.2317927628191035e-06,
          -1.9207778168970373,
          -0.0020420175906035423,
          1.6081048203472623,
          -0.7852961923825373
        ],
        "ee_pos": [
          0.4920747371700454,
          -0.0004221765330002042,
          0.36386578664017494
        ],
        "mug_pos": [
          0.5000255571083206,
          8.085437719119204e-06,
          0.28907062462379196
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0591686951558088e-05,
        -0.010042265066455277,
        -2.047570961272704e-06,
        -1.8936317971112637,
        3.1107695206272384e-05,
        1.6134979660120843,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -8.242758824418416e-06,
          0.018349394709351367,
          -1.3630642201482466e-05,
          -1.9224473170701248,
          -0.0020365833600567975,
          1.6091378594431442,
          -0.7852962771393869
        ],
        "ee_pos": [
          0.4919455575188806,
          -0.000424668846425256,
          0.36127565854556143
        ],
        "mug_pos": [
          0.49999118055590125,
          2.1672971659033668e-05,
          0.28907186511950317
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0586519836738718e-05,
        -0.006111343927414905,
        -2.0420207717259354e-06,
        -1.8953022871808949,
        3.1089875284990585e-05,
        1.6144662411168083,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -9.044105525565928e-06,
          0.02236789798747718,
          -1.947559046595434e-05,
          -1.924101654353165,
          -0.0020320300313188275,
          1.6102194357265462,
          -0.7852962918451392
        ],
        "ee_pos": [
          0.4918206066919707,
          -0.0004268201736311706,
          0.3586512362129604
        ],
        "mug_pos": [
          0.5000273149447866,
          6.49529449328545e-06,
          0.28907260300081133
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0580851013164374e-05,
        -0.0020631590576315494,
        -2.036129400208384e-06,
        -1.8969873913074458,
        3.1072206111558495e-05,
        1.6154895648132461,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -9.378658430526856e-06,
          0.02648091375401108,
          -2.4832578968474964e-05,
          -1.9257501801758115,
          -0.0020264344321466268,
          1.6113417478571206,
          -0.7852962131846204
        ],
        "ee_pos": [
          0.4916945169818221,
          -0.00042829047006286597,
          0.3559854318946779
        ],
        "mug_pos": [
          0.5000259786570651,
          8.245197523767028e-06,
          0.2890707889861554
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.057481599499736e-05,
        0.002093265318702669,
        -2.030047516883824e-06,
        -1.8986805599979009,
        3.1054718933704606e-05,
        1.6165676268388365,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -9.790470680449358e-06,
          0.030278398704428357,
          -3.02917988145128e-05,
          -1.926288871399056,
          -0.002145213911863877,
          1.614294803328494,
          -0.7852977427939141
        ],
        "ee_pos": [
          0.492204101990666,
          -0.0004539672138646864,
          0.35386430909746025
        ],
        "mug_pos": [
          0.4997428931685256,
          1.3457839499909118e-05,
          0.28847855792737476
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0568482864602314e-05,
        0.006346850615951971,
        -2.0238568805506537e-06,
        -1.9003747576715821,
        3.103747668996164e-05,
        1.6176993095731058,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.193446488053621e-05,
          0.03307193759984891,
          -3.813323494954375e-05,
          -1.9257409760766875,
          -0.002155079824167425,
          1.6179605190049802,
          -0.7853036036825651
        ],
        "ee_pos": [
          0.4932595221817542,
          -0.0004618839387113385,
          0.3527245323439472
        ],
        "mug_pos": [
          0.4994466804003231,
          6.3024214170208955e-06,
          0.2879043198319795
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0561889974713706e-05,
        0.010685203157561886,
        -2.0176076210711016e-06,
        -1.9020629068649775,
        3.1020552873625114e-05,
        1.6188827767060228,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.0313155502774218e-05,
          0.03546196059654329,
          -5.3536801926494106e-05,
          -1.9250224670110796,
          -0.002068630834716667,
          1.6200161608427215,
          -0.7852905579732274
        ],
        "ee_pos": [
          0.49399377166850356,
          -0.00045518761639514783,
          0.35185827744290377
        ],
        "mug_pos": [
          0.49902755977933816,
          5.474390796337111e-06,
          0.2875467148904318
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0555063247149339e-05,
        0.015095003959663449,
        -2.0113352088509233e-06,
        -1.9037380995683992,
        3.1004021866908224e-05,
        1.6201155387587374,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.7331778164002446e-05,
          0.037860240734952456,
          -6.615098724839815e-05,
          -1.9244007170866069,
          -0.001995055781588865,
          1.6216363471246749,
          -0.7852868991233805
        ],
        "ee_pos": [
          0.4945877903168892,
          -0.00044818276876707673,
          0.3509493360579764
        ],
        "mug_pos": [
          0.4985489522861354,
          5.598442015779553e-06,
          0.2871484623713271
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0548024099657724e-05,
        0.01956222782707477,
        -2.0050681136256667e-06,
        -1.905393704980884,
        3.0987954462075246e-05,
        1.6213944999650296,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.784004472929533e-05,
          0.04047148207559289,
          -6.959998969109433e-05,
          -1.9239190022526382,
          -0.001967796517019326,
          1.6237464458316542,
          -0.7852954362412151
        ],
        "ee_pos": [
          0.49524722005611416,
          -0.00044486725868519603,
          0.34988505148356336
        ],
        "mug_pos": [
          0.4980691012361138,
          4.279701440582448e-06,
          0.28667101751317575
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0540793072936294e-05,
        0.02407227950731685,
        -1.9988312115946405e-06,
        -1.9070234308563163,
        3.0972415730047705e-05,
        1.6227159933255257,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.9332775111091465e-05,
          0.04328137175012025,
          -6.103130966119843e-05,
          -1.9234610101911012,
          -0.0019662069332345112,
          1.6261937275006009,
          -0.7853053246206009
        ],
        "ee_pos": [
          0.4959724338198216,
          -0.0004371576927333392,
          0.3487173771023798
        ],
        "mug_pos": [
          0.49734214133414245,
          1.085483674316131e-05,
          0.2861513324266382
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0533391481456377e-05,
        0.028610090305368475,
        -1.992647260002874e-06,
        -1.9086213631443898,
        3.095746396651076e-05,
        1.6240758097931223,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.1775477571613352e-05,
          0.046223332418545024,
          -5.4764832857532485e-05,
          -1.9230198639656177,
          -0.0019984289591384267,
          1.6285498998695862,
          -0.7852992874835366
        ],
        "ee_pos": [
          0.49667272240430294,
          -0.00043600869475735244,
          0.34747384890403105
        ],
        "mug_pos": [
          0.49657580949026936,
          8.02661369795544e-06,
          0.2855629765101614
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0525842152770998e-05,
        0.033160196769703595,
        -1.9865375035177054e-06,
        -1.9101819942233684,
        3.09431501533206e-05,
        1.625469225972887,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -5.785149978337841e-06,
          0.04921859287999935,
          -5.0342947214592156e-05,
          -1.9225509988693519,
          -0.001995512623474302,
          1.6308077855291854,
          -0.785297872175537
        ],
        "ee_pos": [
          0.49736399829165195,
          -0.0004304566630308048,
          0.34620061302067306
        ],
        "mug_pos": [
          0.49579586239766055,
          8.986957421973703e-06,
          0.28497910200418225
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0518169739779437e-05,
        0.03770681223178188,
        -1.980521898364678e-06,
        -1.9117002437719508,
        3.092951769095533e-05,
        1.6268910332601763,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.312567970722286e-06,
          0.05221944475014397,
          -4.942438799527033e-05,
          -1.9220548626554992,
          -0.0019786055660065637,
          1.633025418735317,
          -0.7853007453359985
        ],
        "ee_pos": [
          0.498052686298109,
          -0.0004241796567674642,
          0.34493136213304554
        ],
        "mug_pos": [
          0.4948771000133783,
          9.894524915429688e-06,
          0.28442268525467623
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0510400823819442e-05,
        0.04223389652858465,
        -1.974619174169903e-06,
        -1.9131714737169383,
        3.091660229848139e-05,
        1.6283355702078186,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -5.417849226938142e-07,
          0.05520910053520334,
          -5.097137000480811e-05,
          -1.9215410229999517,
          -0.0019729049366196973,
          1.635163544473218,
          -0.7853023963337366
        ],
        "ee_pos": [
          0.49872759030604663,
          -0.0004213905866892827,
          0.3436710113714604
        ],
        "mug_pos": [
          0.4938177243820978,
          7.87271830502599e-06,
          0.28389200197557396
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0502563908519952e-05,
        0.04672522644807839,
        -1.968846835348478e-06,
        -1.9145914976786567,
        3.090443203670631e-05,
        1.6297967590978102,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          1.736432981371007e-05,
          0.05819098578898027,
          -2.7918881841662788e-05,
          -1.9210208978485093,
          -0.0018154860792023834,
          1.6375334575505653,
          -0.7853217674327662
        ],
        "ee_pos": [
          0.4994491927995325,
          -0.00037787827954109134,
          0.3424654129094615
        ],
        "mug_pos": [
          0.4925416182004596,
          -4.363321365684545e-05,
          0.2837371418402457
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.049468934776091e-05,
        0.05116446798303559,
        -1.9632211467291356e-06,
        -1.9159565850037104,
        3.089302743590533e-05,
        1.6312681471264885,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          4.506389962911579e-05,
          0.06349463279117812,
          7.659127317681218e-06,
          -1.9249335573396167,
          -0.0012443781701623362,
          1.636076702177701,
          -0.7853193714772059
        ],
        "ee_pos": [
          0.4982073971825638,
          -0.00022972266853275177,
          0.3385160396183661
        ],
        "mug_pos": [
          0.4921621921763071,
          -3.164135932079826e-05,
          0.28686710085997447
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0486809226169485e-05,
        0.05553525070391168,
        -1.957757122391625e-06,
        -1.9172634594224465,
        3.0882401719151305e-05,
        1.632742952224329,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          7.57266357006965e-05,
          0.07224922473442671,
          4.192168231649404e-05,
          -1.932355892965358,
          -0.0013788894120930022,
          1.6294228709573693,
          -0.7853034244277264
        ],
        "ee_pos": [
          0.49446367553213455,
          -0.0002147310481955999,
          0.3313179245858731
        ],
        "mug_pos": [
          0.4920736165217123,
          -4.3832785236348976e-05,
          0.2887441351704032
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.047895720219405e-05,
        0.059821244139304904,
        -1.95246852737922e-06,
        -1.9185092924231262,
        3.0872561114224995e-05,
        1.6342141132669314,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          8.917915613539779e-05,
          0.08226724105872858,
          4.552377890531215e-05,
          -1.9395438420185096,
          -0.0020362303699153345,
          1.6290780984310431,
          -0.7852886367069847
        ],
        "ee_pos": [
          0.49192880820864426,
          -0.0003355688647637461,
          0.32356029495851923
        ],
        "mug_pos": [
          0.49174763070573824,
          2.364065881230014e-07,
          0.2885278943144584
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0471168317237969e-05,
        0.06400623581779719,
        -1.947367894399166e-06,
        -1.919691691523583,
        3.086350524816796e-05,
        1.635674344253108,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          6.803388536653512e-05,
          0.09146178980708446,
          7.613113413210485e-07,
          -1.9450873907367985,
          -0.0023913745493644,
          1.6326269219768097,
          -0.7852882060423277
        ],
        "ee_pos": [
          0.49078937127132594,
          -0.0004426839260815353,
          0.3168420542951729
        ],
        "mug_pos": [
          0.49160572003836117,
          -8.195292141616811e-06,
          0.2883230134700197
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0463478774010763e-05,
        0.06807421049303142,
        -1.9424665562957575e-06,
        -1.9208086837149383,
        3.085522761726778e-05,
        1.6371161919028923,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          2.789910367229253e-05,
          0.09885351757860186,
          -6.431703879089771e-05,
          -1.9487115929762933,
          -0.0022818846932474127,
          1.6349579532102303,
          -0.7852988132905343
        ],
        "ee_pos": [
          0.489961672721657,
          -0.0004769563788135593,
          0.3117294011337025
        ],
        "mug_pos": [
          0.4914715077458161,
          -6.93426156602519e-06,
          0.28852256400854215
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0455925685994567e-05,
        0.07200943000077438,
        -1.937774692872206e-06,
        -1.921858694439613,
        3.084771612462905e-05,
        1.6385320960456458,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -3.6153776644077646e-06,
          0.1043359968807154,
          -0.000115007285310285,
          -1.9508526883532586,
          -0.0020470647499923884,
          1.6359082752620875,
          -0.7853038266928274
        ],
        "ee_pos": [
          0.48934209069742546,
          -0.0004679250525769671,
          0.30816343545683733
        ],
        "mug_pos": [
          0.49111474030749985,
          -0.00010797921849198724,
          0.28865647857194765
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0448546801219502e-05,
        0.07579651315897239,
        -1.933301390997247e-06,
        -1.9228405225435927,
        3.0840953676079516e-05,
        1.639914452115218,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.4334491685314458e-06,
          0.10841719409431978,
          -0.00011641326707851441,
          -1.9519077581526634,
          -0.001857403059177039,
          1.6374117124616459,
          -0.7853124334606388
        ],
        "ee_pos": [
          0.48920481142153,
          -0.00043231065596954934,
          0.3057166069533733
        ],
        "mug_pos": [
          0.4906026968520006,
          -0.0001935974597662314,
          0.2888153539780317
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0441380203533777e-05,
        0.0794205151066511,
        -1.929054716126296e-06,
        -1.923753311709429,
        3.0834918824516814e-05,
        1.641255675038197,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          9.070601573819907e-06,
          0.11141730787409498,
          -0.00010765821695214369,
          -1.9525860440375022,
          -0.0019211029372452608,
          1.63816731317086,
          -0.7852953645492082
        ],
        "ee_pos": [
          0.4890661748887335,
          -0.0004323049131508956,
          0.30395544709224576
        ],
        "mug_pos": [
          0.49054090462447997,
          -0.00021159455298987994,
          0.2889342390589559
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0437600980981308e-05,
        0.08142169248695778,
        -1.926943469204054e-06,
        -1.9242887184112474,
        3.08309864753479e-05,
        1.6419730195406923,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          2.399172335722841e-05,
          0.11399224332921486,
          -9.257763784267925e-05,
          -1.9528594131390822,
          -0.0019465513825544332,
          1.639465063126439,
          -0.7852893916158652
        ],
        "ee_pos": [
          0.4891747991757389,
          -0.0004243466994312415,
          0.3025805556237557
        ],
        "mug_pos": [
          0.4905706816507755,
          -0.0002290007825605909,
          0.28892248413717825
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0429936604328196e-05,
        0.08536008498198323,
        -1.9226812950658054e-06,
        -1.9252915666880444,
        3.082413007234136e-05,
        1.643420918408651,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          2.4861832958286314e-05,
          0.11643576323833893,
          -9.588579104126036e-05,
          -1.9533335008335901,
          -0.001914429865835249,
          1.6400985006827442,
          -0.785294826428255
        ],
        "ee_pos": [
          0.48908590098450605,
          -0.00041822292768416044,
          0.3012012032126947
        ],
        "mug_pos": [
          0.4905274165757026,
          -0.0001476545497550138,
          0.28905429443759745
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0426089756707864e-05,
        0.08733654340389824,
        -1.9206017786073296e-06,
        -1.9257919843672096,
        3.0820717941755484e-05,
        1.64414912032328,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          2.188919478522747e-05,
          0.11903082377274943,
          -0.00010304818253741148,
          -1.953730597138215,
          -0.0018910183226338046,
          1.6413642257244148,
          -0.7852905603902461
        ],
        "ee_pos": [
          0.4891464216600399,
          -0.000418189260627382,
          0.299789126746424
        ],
        "mug_pos": [
          0.4903642235671844,
          -0.00023439279867909927,
          0.2889966537920877
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.041859476393544e-05,
        0.09103094415295453,
        -1.9165189642278522e-06,
        -1.9266643348229633,
        3.081545298176037e-05,
        1.6455552283770802,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          1.655693509209553e-05,
          0.1216711504748555,
          -0.0001129448940697719,
          -1.9543001747396644,
          -0.0018877340261187565,
          1.6423986069257297,
          -0.7852922999029948
        ],
        "ee_pos": [
          0.4890971893401025,
          -0.00042466584766100233,
          0.2982752691136667
        ],
        "mug_pos": [
          0.4904178118343374,
          -0.00019173394852955303,
          0.289058032527883
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.041497253338486e-05,
        0.09280327327767163,
        -1.9145839227058005e-06,
        -1.9270761251872446,
        3.081303142735694e-05,
        1.6462341910498288,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          1.3407350203195463e-05,
          0.12410040756968063,
          -0.00012114273312878026,
          -1.9548502815396493,
          -0.00192024294242687,
          1.6431062204840323,
          -0.7852919736815436
        ],
        "ee_pos": [
          0.4889959778745421,
          -0.0004357023779087572,
          0.2968747063405373
        ],
        "mug_pos": [
          0.4902181677963885,
          -0.00024163348980242003,
          0.28902972536958305
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0411236968408316e-05,
        0.0946644527161092,
        -1.9126533185605196e-06,
        -1.9275197259249646,
        3.0810269426375305e-05,
        1.6469387704928258,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          1.0911628119223156e-05,
          0.126283994478677,
          -0.000127563033703237,
          -1.9552898028362986,
          -0.001927698528701553,
          1.6438845622357052,
          -0.7852924247777832
        ],
        "ee_pos": [
          0.48893834584868945,
          -0.00044217203418870363,
          0.2956290955444387
        ],
        "mug_pos": [
          0.4902021117030358,
          -0.00021153519379215704,
          0.28899198729256903
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0407490589788305e-05,
        0.09651711950188402,
        -1.910738536400563e-06,
        -1.9279543141848017,
        3.0807633677174865e-05,
        1.6476448907091623,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          3.2301413839150866e-06,
          0.12825275816716492,
          -0.0001401938830268954,
          -1.9556684925827907,
          -0.0019055432531705364,
          1.6445260751612731,
          -0.7853068218109843
        ],
        "ee_pos": [
          0.48887815053754213,
          -0.0004477061471766767,
          0.2945164407012768
        ],
        "mug_pos": [
          0.49001257356710254,
          -0.00021472633498705165,
          0.28893316677782
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.040382859832852e-05,
        0.09828962282054451,
        -1.9088621801913787e-06,
        -1.9283541212191684,
        3.080539172731909e-05,
        1.6483317603734944,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.2164509543765007e-06,
          0.13006724721139956,
          -0.0001490742767978561,
          -1.9560303594422377,
          -0.001876361626312081,
          1.6451571879633111,
          -0.7853060580223764
        ],
        "ee_pos": [
          0.4888256351712663,
          -0.00044895389307408076,
          0.2934871238652035
        ],
        "mug_pos": [
          0.48996826177254227,
          -0.000213994172622032,
          0.28894276830877424
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0400337154650766e-05,
        0.0999287654378955,
        -1.9070514988951653e-06,
        -1.9287033036970642,
        3.0803684024144804e-05,
        1.6489815885873436,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -8.910504808084919e-06,
          0.13146248882142078,
          -0.00016156043368575104,
          -1.956339259943687,
          -0.0019433177338834397,
          1.645460834239467,
          -0.7852914899140224
        ],
        "ee_pos": [
          0.48873352191669933,
          -0.00047016942676743686,
          0.29266860051024657
        ],
        "mug_pos": [
          0.4897723719304311,
          -0.00022524144980910692,
          0.28891962647691555
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0400337154650766e-05,
        0.0999287654378955,
        -1.9070514988951653e-06,
        -1.9287033036970642,
        3.0803684024144804e-05,
        1.6489815885873436,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.7002535392625386e-05,
          0.13257422570788607,
          -0.00017396848499725152,
          -1.9565111380814357,
          -0.0019573201976676993,
          1.6457072825814973,
          -0.7852905150794994
        ],
        "ee_pos": [
          0.4886759094426914,
          -0.0004844796484738714,
          0.2920542599909035
        ],
        "mug_pos": [
          0.48975109785676907,
          -0.00019801735194302728,
          0.2889159964690636
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.2662380457063895e-05,
          0.133430522027523,
          -0.00018270586552250866,
          -1.956694776519178,
          -0.001917652878035183,
          1.6458762393143733,
          -0.7852945433887649
        ],
        "ee_pos": [
          0.4886226741224424,
          -0.00048385416388226483,
          0.29155910348900504
        ],
        "mug_pos": [
          0.4896013814094388,
          -0.00020088580913218973,
          0.28893860573981933
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.417480348719264e-05,
          0.1339639566590616,
          -0.00018552560010546494,
          -1.956825532244312,
          -0.0019081731423046414,
          1.6458524199151352,
          -0.7852964546964374
        ],
        "ee_pos": [
          0.48854592114103307,
          -0.00048401213542319455,
          0.29123824979063384
        ],
        "mug_pos": [
          0.4895046150247533,
          -0.00022031799757137884,
          0.2889700083161637
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.3259662286288913e-05,
          0.13421985598802338,
          -0.00018528627081905845,
          -1.956907422477319,
          -0.0019129926862318078,
          1.6457123167492178,
          -0.7852977607636011
        ],
        "ee_pos": [
          0.4884752072746293,
          -0.0004841165890499959,
          0.29107683786441585
        ],
        "mug_pos": [
          0.48947614014161905,
          -0.00019357089706985943,
          0.2890203930591104
        ],
        "gripper_state": 1.0
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.04
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.6777369825556783e-05,
          0.13432998198871535,
          -0.0001911634990159286,
          -1.956949981475054,
          -0.0019486878763327793,
          1.645738848535392,
          -0.7852869083637306
        ],
        "ee_pos": [
          0.48845865856785536,
          -0.0004957923684727054,
          0.2910039857619509
        ],
        "mug_pos": [
          0.4893655754568097,
          -0.00018477690317848675,
          0.28912308331917175
        ],
        "gripper_state": 0.9583333333333334
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.03833333333333334
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.5813080846183235e-05,
          0.13437707044291958,
          -0.0001896281167358305,
          -1.9569365126242797,
          -0.001910396142428859,
          1.6459595155943727,
          -0.7852920845991153
        ],
        "ee_pos": [
          0.4885071834896554,
          -0.0004874602253736169,
          0.29098168890135134
        ],
        "mug_pos": [
          0.48921571925346913,
          -0.00019210919365447132,
          0.289255305023002
        ],
        "gripper_state": 0.9166666666666666
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.03666666666666667
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.269699495728313e-05,
          0.1344175755714973,
          -0.00018546004391664796,
          -1.9569026927910773,
          -0.0018966804651575878,
          1.6462297287869068,
          -0.7852914471281703
        ],
        "ee_pos": [
          0.4885781542214449,
          -0.0004810476174327397,
          0.29097200767342857
        ],
        "mug_pos": [
          0.48900688626440175,
          -0.00018812862355452935,
          0.2893996635766845
        ],
        "gripper_state": 0.875
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.035
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.0038183640546834e-05,
          0.13447958913533817,
          -0.00018212196350002294,
          -1.9568817344001272,
          -0.0019051430894779634,
          1.6464342202977384,
          -0.7852907609890707
        ],
        "ee_pos": [
          0.48863056401892957,
          -0.00047972459166925775,
          0.2909486343081817
        ],
        "mug_pos": [
          0.488675188963546,
          -0.00018999221314469142,
          0.2895119942522545
        ],
        "gripper_state": 0.8333333333333334
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.03333333333333333
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.719797108698957e-05,
          0.13456547742144492,
          -0.00017844355066624142,
          -1.9568806597690611,
          -0.0019025540643315723,
          1.6465625207125427,
          -0.7852922232238155
        ],
        "ee_pos": [
          0.4886571958124169,
          -0.00047608869320047337,
          0.2909065231587121
        ],
        "mug_pos": [
          0.48817442193141486,
          -0.00018719107761932442,
          0.289632208557989
        ],
        "gripper_state": 0.7916666666666666
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.03166666666666666
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.4596589755562283e-05,
          0.13465826865220987,
          -0.00017514829475711276,
          -1.956894217838779,
          -0.0018981478605511616,
          1.6466474718455402,
          -0.7852925569191908
        ],
        "ee_pos": [
          0.48866843813787897,
          -0.0004722663605949128,
          0.29085558997339717
        ],
        "mug_pos": [
          0.48751241149995,
          -0.0001852262638570076,
          0.28982305792773927
        ],
        "gripper_state": 0.75
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.03
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.2757277261598255e-05,
          0.13471665135042987,
          -0.00017290102812441075,
          -1.9569758073663892,
          -0.001896177766480274,
          1.646363379634618,
          -0.7852932990697926
        ],
        "ee_pos": [
          0.4885870464715233,
          -0.0004698247194216537,
          0.29080121502768513
        ],
        "mug_pos": [
          0.4869750669120877,
          -0.00018351601851705854,
          0.2898344608153513
        ],
        "gripper_state": 0.7083333333333333
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.028333333333333332
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.1337228747774312e-05,
          0.13472377986838543,
          -0.0001712340151557328,
          -1.9571169039509317,
          -0.0019009842734543348,
          1.6458039012377266,
          -0.7852927586424294
        ],
        "ee_pos": [
          0.48840682272848224,
          -0.00046923997573122106,
          0.29074939918357157
        ],
        "mug_pos": [
          0.48683531848763467,
          -0.0001827495138134012,
          0.289754925346113
        ],
        "gripper_state": 0.6666666666666667
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.026666666666666672
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.054733891109252e-05,
          0.13467892838706413,
          -0.00017033630218498498,
          -1.957204196539583,
          -0.001904725986960263,
          1.6456069816883125,
          -0.7852938436798479
        ],
        "ee_pos": [
          0.48832905666744886,
          -0.00046920224195913193,
          0.2907390189484211
        ],
        "mug_pos": [
          0.4867528055737378,
          -0.00018289151173138393,
          0.28973765663559786
        ],
        "gripper_state": 0.625
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.025
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.1141957647132403e-05,
          0.13460535946423216,
          -0.00017104830112190834,
          -1.9572368906518476,
          -0.0019002607716907928,
          1.6455790803083383,
          -0.7852983862167513
        ],
        "ee_pos": [
          0.4883123042742711,
          -0.00046895291209074914,
          0.2907617553736171
        ],
        "mug_pos": [
          0.4867306357874692,
          -0.0001834177691519392,
          0.2897578927379796
        ],
        "gripper_state": 0.5833333333333333
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.02333333333333333
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.180725336537734e-05,
          0.13452779807077164,
          -0.0001717521342100289,
          -1.9572280755964966,
          -0.0018964417666867534,
          1.6455934125581326,
          -0.7852971917819997
        ],
        "ee_pos": [
          0.48832085496095595,
          -0.00046884502008350307,
          0.2908020600441359
        ],
        "mug_pos": [
          0.48673827390957886,
          -0.00018372490141384118,
          0.2897976705534976
        ],
        "gripper_state": 0.5416666666666667
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.02166666666666667
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.2112785780098372e-05,
          0.13445709408078038,
          -0.00017198574429907044,
          -1.957194241461486,
          -0.00189547538144395,
          1.6456202967293838,
          -0.7852954295197491
        ],
        "ee_pos": [
          0.4883410041079581,
          -0.0004689057853270814,
          0.2908484638795874
        ],
        "mug_pos": [
          0.4867601706710067,
          -0.0001841272291820721,
          0.28984681046248484
        ],
        "gripper_state": 0.5
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.02
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.2167838758747328e-05,
          0.13439382272257455,
          -0.00017193506259503483,
          -1.9571453545959925,
          -0.00189607376961291,
          1.6456548341587776,
          -0.7852957139477085
        ],
        "ee_pos": [
          0.48836792278031793,
          -0.00046902531724341614,
          0.290896740146459
        ],
        "mug_pos": [
          0.4867903122649715,
          -0.00018460273677216757,
          0.2899027137133158
        ],
        "gripper_state": 0.45833333333333337
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.018333333333333337
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.183023128604969e-05,
          0.1343346456371993,
          -0.00017144925190613094,
          -1.9570861826219772,
          -0.001897879621009316,
          1.6456943044414576,
          -0.7852940794818433
        ],
        "ee_pos": [
          0.4883994098806374,
          -0.0004689991338290652,
          0.29094668034778887
        ],
        "mug_pos": [
          0.48682611603172005,
          -0.00018451877957360028,
          0.28996607234547156
        ],
        "gripper_state": 0.41666666666666663
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.016666666666666666
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.1077127162620839e-05,
          0.1342764227907846,
          -0.00017048459756471948,
          -1.9570195804263755,
          -0.0019003403581497199,
          1.6457373098770598,
          -0.7852918648977159
        ],
        "ee_pos": [
          0.48843426098144854,
          -0.00046866562046460716,
          0.29099882390477444
        ],
        "mug_pos": [
          0.4868663164962383,
          -0.00018404677040746492,
          0.2900376062202961
        ],
        "gripper_state": 0.375
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.015
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.055555595900507e-05,
          0.1342177550379411,
          -0.00016981112539626373,
          -1.956947681446287,
          -0.0019006781473340556,
          1.6457832966297428,
          -0.7852927377731308
        ],
        "ee_pos": [
          0.48847166073318277,
          -0.0004681472851663397,
          0.2910531054220701
        ],
        "mug_pos": [
          0.48691008948116427,
          -0.00018351566178240745,
          0.2901172479184606
        ],
        "gripper_state": 0.33333333333333337
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.013333333333333336
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0420874161891779e-05,
          0.13415822861357596,
          -0.00016959805995855721,
          -1.9568719182865335,
          -0.0018997154541555564,
          1.6458316143696219,
          -0.7852937452866827
        ],
        "ee_pos": [
          0.4885109800695982,
          -0.0004677646323779605,
          0.2911092225779177
        ],
        "mug_pos": [
          0.4869567720078524,
          -0.0001832131470833581,
          0.2902047375951705
        ],
        "gripper_state": 0.29166666666666663
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.011666666666666665
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0498154117319948e-05,
          0.13409785694583723,
          -0.00016961146422220718,
          -1.9567933379801437,
          -0.0018987818772439292,
          1.6458816387681523,
          -0.7852937834393041
        ],
        "ee_pos": [
          0.4885517006751255,
          -0.00046761280116855013,
          0.2911667983331616
        ],
        "mug_pos": [
          0.48700591873482646,
          -0.00018311125670710215,
          0.2902997635053605
        ],
        "gripper_state": 0.25
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.01
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0530697648014764e-05,
          0.13403685900657816,
          -0.00016954760615233542,
          -1.9567127686088261,
          -0.0018982200648164234,
          1.6459330130278875,
          -0.7852939900127619
        ],
        "ee_pos": [
          0.4885934581076886,
          -0.0004674787658016239,
          0.2912254221742495
        ],
        "mug_pos": [
          0.4870571764663359,
          -0.00018300003981642693,
          0.290401969906864
        ],
        "gripper_state": 0.20833333333333337
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.008333333333333335
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0493069537339299e-05,
          0.13397541492580792,
          -0.00016940356657924976,
          -1.9566307973919719,
          -0.0018983045795340917,
          1.6459851049783811,
          -0.785294067214608
        ],
        "ee_pos": [
          0.48863588294730953,
          -0.00046740035981163267,
          0.2912847970382968
        ],
        "mug_pos": [
          0.4871104437479822,
          -0.0001829479792103896,
          0.29051111311520605
        ],
        "gripper_state": 0.16666666666666663
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.006666666666666665
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0433425711403884e-05,
          0.13391376454950046,
          -0.00016924281491414222,
          -1.956547921281154,
          -0.0018985870337350302,
          1.646037910887857,
          -0.785293984451823
        ],
        "ee_pos": [
          0.4886787914452044,
          -0.0004673445894198265,
          0.29134462382811993
        ],
        "mug_pos": [
          0.487165508539996,
          -0.00018292714539150555,
          0.2906269045311647
        ],
        "gripper_state": 0.125
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.005
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0396256740655086e-05,
          0.13385216030626174,
          -0.0001691178679189024,
          -1.956464544557985,
          -0.001898777314452003,
          1.6460912917193853,
          -0.7852939020749863
        ],
        "ee_pos": [
          0.4887220024661084,
          -0.0004672983605326014,
          0.291404631210375
        ],
        "mug_pos": [
          0.48722227234195425,
          -0.0001829272978067508,
          0.29074905824024694
        ],
        "gripper_state": 0.08333333333333337
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.003333333333333335
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0396107376523815e-05,
          0.13379084347798775,
          -0.00016904257494780342,
          -1.9563810111502282,
          -0.0018988826952186814,
          1.6461450658914007,
          -0.7852938523682127
        ],
        "ee_pos": [
          0.4887653457701453,
          -0.0004672764910549692,
          0.2914645761715221
        ],
        "mug_pos": [
          0.4872806699777871,
          -0.00018296045948398075,
          0.29087729339194673
        ],
        "gripper_state": 0.04166666666666663
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.0016666666666666653
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0426912213803489e-05,
          0.13373003857294857,
          -0.00016900765247016185,
          -1.9562976295801635,
          -0.0018989939879272227,
          1.6461990775693767,
          -0.7852937911866057
        ],
        "ee_pos": [
          0.4888086699304327,
          -0.00046729037543159313,
          0.2915242367368071
        ],
        "mug_pos": [
          0.48734066327707176,
          -0.00018303578322907856,
          0.2910113280835243
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0500919707795445e-05,
          0.1336702042714713,
          -0.00016902340301558341,
          -1.9562194991211146,
          -0.0018988006108912275,
          1.646235392327049,
          -0.7852937455711488
        ],
        "ee_pos": [
          0.48884661563867116,
          -0.00046729752131964753,
          0.2915819473208952
        ],
        "mug_pos": [
          0.4873984098878767,
          -0.00018311695251158294,
          0.29114701559184064
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0610630089765528e-05,
          0.13361616005715427,
          -0.00016907926729417894,
          -1.956152636562049,
          -0.001898295203565153,
          1.6462557318975122,
          -0.7852936724716592
        ],
        "ee_pos": [
          0.4888768680265804,
          -0.00046727669229755156,
          0.29163308922211406
        ],
        "mug_pos": [
          0.487448820697056,
          -0.0001831842770716598,
          0.29127988344766914
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0702132748141996e-05,
          0.1335703143712912,
          -0.00016911625390603717,
          -1.9560981377302418,
          -0.001898170805990488,
          1.6462661998954005,
          -0.7852936334505611
        ],
        "ee_pos": [
          0.4889002190489739,
          -0.0004673111862716152,
          0.2916758172876665
        ],
        "mug_pos": [
          0.4874926538452456,
          -0.00018330547695487748,
          0.2914081462303823
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0752502135302783e-05,
          0.13353223790478555,
          -0.0001691160359340379,
          -1.9560543779268666,
          -0.0018985503363726263,
          1.646270800900217,
          -0.7852936255012294
        ],
        "ee_pos": [
          0.4889181524620552,
          -0.0004674105409043051,
          0.2917108056185368
        ],
        "mug_pos": [
          0.48753148292065995,
          -0.00018348898456463456,
          0.29153235020519946
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0786901844937225e-05,
          0.13350064928510172,
          -0.00016911302910198724,
          -1.9560194488572293,
          -0.0018990591027991536,
          1.6462715985155165,
          -0.7852935951028078
        ],
        "ee_pos": [
          0.4889318652808301,
          -0.0004675292832354229,
          0.2917393586891919
        ],
        "mug_pos": [
          0.487566634033497,
          -0.00018369765766686799,
          0.29165331049199644
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.083659139450238e-05,
          0.13347452001251997,
          -0.0001691422364279951,
          -1.9559918562594925,
          -0.0018994245307201512,
          1.6462698330966565,
          -0.785293543720016
        ],
        "ee_pos": [
          0.48894221137636246,
          -0.0004676429398035601,
          0.29176253372485395
        ],
        "mug_pos": [
          0.48759904855521835,
          -0.0001839110869286467,
          0.2917714992917473
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0396831852958673e-05,
        0.10159406134185749,
        -1.905281114984238e-06,
        -1.9290648423027166,
        3.0801813689436405e-05,
        1.6496365736065084,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.114523853134645e-05,
          0.13295896942770735,
          -0.0001700221717207797,
          -1.9559564522412884,
          -0.001926895386422995,
          1.6458029485115824,
          -0.7852949459544739
        ],
        "ee_pos": [
          0.4888769474355743,
          -0.00047349817572244564,
          0.2920084069309891
        ],
        "mug_pos": [
          0.48757057979651086,
          -0.00018875778993554442,
          0.2920356740795447
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0399640389626473e-05,
        0.09801592568777226,
        -1.9043433235238853e-06,
        -1.9274274998732697,
        3.082165461272646e-05,
        1.6488548206260185,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.3733157601250903e-05,
          0.13157085397224155,
          -0.00017395564723976276,
          -1.95588351340511,
          -0.0019670875800587634,
          1.6443652760801133,
          -0.7852937379758986
        ],
        "ee_pos": [
          0.4886456409199139,
          -0.00048465099795295616,
          0.2926954695129661
        ],
        "mug_pos": [
          0.48735071520797246,
          -0.000199656861707986,
          0.29261970689441996
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0404897286489854e-05,
        0.09410287011918411,
        -1.9056438165049897e-06,
        -1.9260296822263088,
        3.083607994353947e-05,
        1.647714975761998,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.0150687573804247e-05,
          0.12916049822360243,
          -0.0001818790900513947,
          -1.9557345837563942,
          -0.0019930763218668863,
          1.6416924249680969,
          -0.7852924587306132
        ],
        "ee_pos": [
          0.4881994604818265,
          -0.0004975515875817686,
          0.2939180584334018
        ],
        "mug_pos": [
          0.4868596321829952,
          -0.00021273096619358047,
          0.293566316393887
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0411988381840052e-05,
        0.08970048504928171,
        -1.9084607526350167e-06,
        -1.9246576569330953,
        3.084853687377021e-05,
        1.646286359563066,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.7547493306172504e-05,
          0.12575972059765225,
          -0.00018956296996494183,
          -1.9553141279779922,
          -0.0019965039295644373,
          1.6391117940423259,
          -0.7852918209833206
        ],
        "ee_pos": [
          0.48787072070071424,
          -0.0005063115971298837,
          0.2957180441150614
        ],
        "mug_pos": [
          0.48652762331271965,
          -0.00022166033406162687,
          0.2950594599730497
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0420547653997521e-05,
        0.08475420518851895,
        -1.9124437815682973e-06,
        -1.9232064576598722,
        3.086076417853909e-05,
        1.6446144654031107,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -3.410765860155071e-05,
          0.12111929905295464,
          -0.0001952915283173312,
          -1.9544758469988315,
          -0.002022604251659751,
          1.636854104820492,
          -0.785291836703858
        ],
        "ee_pos": [
          0.4878023961523768,
          -0.0005176751936568512,
          0.29825637071184175
        ],
        "mug_pos": [
          0.48646148954624785,
          -0.00023326937077945985,
          0.2972919873740985
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0433972107891544e-05,
        0.0776027699652745,
        -1.9196885203161037e-06,
        -1.9212707056525027,
        3.0875249266118277e-05,
        1.6420764772250696,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -3.918171353906871e-05,
          0.11526224293334397,
          -0.00019732654959762474,
          -1.9530860399720367,
          -0.0020425643416486132,
          1.6344348182712725,
          -0.7852922239662616
        ],
        "ee_pos": [
          0.4879088959023421,
          -0.0005263114886861009,
          0.3015995837148012
        ],
        "mug_pos": [
          0.48656838792109197,
          -0.00024226241010768954,
          0.300337585740425
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.044706702405457e-05,
        0.07078173214619404,
        -1.9273792675445603e-06,
        -1.9194454265961889,
        3.088845785668813e-05,
        1.6396369884188757,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.1753019824893915e-05,
          0.1084990082144915,
          -0.00019405922734451087,
          -1.9512181499478016,
          -0.0020149457603672693,
          1.6320004606666985,
          -0.785292523996129
        ],
        "ee_pos": [
          0.4881737510682065,
          -0.0005217466413759675,
          0.30557651506701833
        ],
        "mug_pos": [
          0.4868318856358545,
          -0.00023794895315112964,
          0.3040270569866744
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0460216101634119e-05,
        0.06378723525346554,
        -1.935481364898923e-06,
        -1.917490759049433,
        3.090332037957393e-05,
        1.637192492669048,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.023396784215383e-05,
          0.10112364468540674,
          -0.0001839055464997104,
          -1.9490160512520884,
          -0.001974852816231904,
          1.6296149826894188,
          -0.7852931610949262
        ],
        "ee_pos": [
          0.4885452185912191,
          -0.0005087136576241008,
          0.3099910165756863
        ],
        "mug_pos": [
          0.48720082450410274,
          -0.00022503320121807093,
          0.30816440692642705
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.047362764134342e-05,
        0.05646228310380546,
        -1.944145205588657e-06,
        -1.9153426538854972,
        3.092052362594755e-05,
        1.6347028800939771,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -3.485936737271164e-05,
          0.09333919356867565,
          -0.0001679287408158586,
          -1.9466079735326123,
          -0.0019605622633656903,
          1.6271549576294402,
          -0.7852939583306211
        ],
        "ee_pos": [
          0.4889385669120838,
          -0.0004957988363188568,
          0.31469525926238545
        ],
        "mug_pos": [
          0.4875903506105796,
          -0.00021220433793082426,
          0.3126013213772939
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0487361608198912e-05,
        0.04877537306409155,
        -1.953491021513925e-06,
        -1.9129852409068941,
        3.0940230934491824e-05,
        1.632162306995208,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.7890987237125487e-05,
          0.08526296946969253,
          -0.00014950823057952638,
          -1.9440786047188723,
          -0.001968328347547111,
          1.624540953950735,
          -0.7852943660501303
        ],
        "ee_pos": [
          0.4892959002042898,
          -0.00048540228495414525,
          0.31959769907265534
        ],
        "mug_pos": [
          0.4879423703729368,
          -0.0002019123678544239,
          0.3172451344638855
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0501388521359433e-05,
        0.0407423036462664,
        -1.963585951085653e-06,
        -1.910414767187818,
        3.0962523484216694e-05,
        1.6295823626243011,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.1608175219373208e-05,
          0.07695345459029843,
          -0.00013170028120294413,
          -1.9414578448729283,
          -0.0019799998511153016,
          1.6218243295023864,
          -0.7852944090091708
        ],
        "ee_pos": [
          0.48961030239380937,
          -0.00047648812207178157,
          0.32465560311168584
        ],
        "mug_pos": [
          0.4882503468463069,
          -0.000193103178921937,
          0.3220517207965285
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0515632613942535e-05,
        0.03239766009376241,
        -1.974457279994063e-06,
        -1.9076304286040406,
        3.098748789058991e-05,
        1.6269830431247987,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.709548042684957e-05,
          0.06843305706819543,
          -0.00011590427697577246,
          -1.9387296771166356,
          -0.0019903388928289027,
          1.6190993704367274,
          -0.7852944398387942
        ],
        "ee_pos": [
          0.48990253641279435,
          -0.0004691004049840485,
          0.3298626102008533
        ],
        "mug_pos": [
          0.4885358897818769,
          -0.00018581333100673548,
          0.32701361314595434
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0529997487404753e-05,
        0.0237839619048444,
        -1.986105894173394e-06,
        -1.904632612125034,
        3.1015217719309385e-05,
        1.62438805769475,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.4505859496695833e-05,
          0.059706029951074636,
          -0.00010231880934025301,
          -1.9358563493842793,
          -0.002002419158138061,
          1.6164224843464763,
          -0.7852945490625399
        ],
        "ee_pos": [
          0.4901926943887437,
          -0.00046405052338967104,
          0.33523148659270846
        ],
        "mug_pos": [
          0.4888201236987564,
          -0.0001808857159764375,
          0.332142933489477
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0544380248037966e-05,
        0.014947254988055901,
        -1.9985146281134987e-06,
        -1.901422942417669,
        3.1045800212000956e-05,
        1.6218223621028633,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.4652273368916272e-05,
          0.050511809729798525,
          -9.215878969076429e-05,
          -1.9328205325699497,
          -0.0020367878722297233,
          1.6134157184383593,
          -0.7852948087119226
        ],
        "ee_pos": [
          0.4904020088702383,
          -0.00046622931476710764,
          0.3408830531972924
        ],
        "mug_pos": [
          0.4890237591461595,
          -0.00018327057565840408,
          0.3375582966486487
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0561765532717935e-05,
        0.0044602572478273685,
        -2.0146449394128297e-06,
        -1.8975903073057994,
        3.108204109766868e-05,
        1.618793334716512,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.6040924107050697e-05,
          0.04111203095618171,
          -8.295833063279758e-05,
          -1.9294863245965712,
          -0.0020383078636120595,
          1.6107080957710957,
          -0.7852942930723026
        ],
        "ee_pos": [
          0.49067881245191586,
          -0.00046407937731123906,
          0.34677772007984364
        ],
        "mug_pos": [
          0.48929830697697446,
          -0.0001812897570395619,
          0.34322353988121584
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0574338393758426e-05,
        -0.003753304384563781,
        -2.0270576088923945e-06,
        -1.894304546168297,
        3.111535009504654e-05,
        1.616628255606827,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.7384555687307175e-05,
          0.03150949661112869,
          -7.330229743852574e-05,
          -1.9259265878088532,
          -0.002035379661252165,
          1.60833386127523,
          -0.7852946886078388
        ],
        "ee_pos": [
          0.4910344855335428,
          -0.0004595491650211597,
          0.35286270492739785
        ],
        "mug_pos": [
          0.48965565194386107,
          -0.00017683620247885623,
          0.3490825087021204
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.059018883300468e-05,
        -0.014075022063903389,
        -2.0437342533891037e-06,
        -1.8900929574632823,
        3.1158272037506564e-05,
        1.6139689549554097,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.702272797673545e-05,
          0.02190926048159226,
          -6.10798011270639e-05,
          -1.9220948744262465,
          -0.0020311581942870035,
          1.6062233938059205,
          -0.7852952627919351
        ],
        "ee_pos": [
          0.4914432015467424,
          -0.0004534252150729004,
          0.3590848715740423
        ],
        "mug_pos": [
          0.49006974717809254,
          -0.0001708175030817093,
          0.3550865303691064
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0601881124111474e-05,
        -0.022287207638599815,
        -2.0569023943873195e-06,
        -1.8864686223032037,
        3.1197148649167546e-05,
        1.6120552176268013,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.5928628839123874e-05,
          0.012244017353117028,
          -4.7705440347957586e-05,
          -1.9181026066495999,
          -0.0020445609252711253,
          1.6040893200000663,
          -0.78529583264767
        ],
        "ee_pos": [
          0.49182301553411917,
          -0.00044903894809149693,
          0.36540426864006187
        ],
        "mug_pos": [
          0.49045714964889264,
          -0.00016654970652257376,
          0.3611916266540007
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.061663831462735e-05,
        -0.03257214152005793,
        -2.0745385467331795e-06,
        -1.8818463823332567,
        3.1246896236793776e-05,
        1.6097229906652175,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.422926579496635e-05,
          0.0026729925255062825,
          -3.3553421659389414e-05,
          -1.9139295926777753,
          -0.00204634118280003,
          1.6021013464286429,
          -0.7852957629801857
        ],
        "ee_pos": [
          0.4921947536399604,
          -0.0004425688054431593,
          0.37178024284147915
        ],
        "mug_pos": [
          0.4908390476857474,
          -0.00016018811817123147,
          0.3673602526225334
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0627476161202224e-05,
        -0.040700704249352036,
        -2.0884617127912474e-06,
        -1.8779244593618332,
        3.129084893185983e-05,
        1.6080807648437072,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.2713540328025251e-05,
          -0.006869816014046786,
          -1.949166621422614e-05,
          -1.9096421193399014,
          -0.0020568656184529062,
          1.6002309063067797,
          -0.785296035598318
        ],
        "ee_pos": [
          0.4925506625353865,
          -0.000436949922527595,
          0.3781921554646076
        ],
        "mug_pos": [
          0.49120766173366376,
          -0.00015465513327905697,
          0.37356731264014165
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0641024517919733e-05,
        -0.05078831819080133,
        -2.1068429034854394e-06,
        -1.8729590668207579,
        3.134673026140973e-05,
        1.6061209117478468,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.1019476926668587e-05,
          -0.016249140584556187,
          -5.4372447781066624e-06,
          -1.905202054433534,
          -0.002058180500973645,
          1.5986299825706303,
          -0.7852961984744248
        ],
        "ee_pos": [
          0.49292104725495134,
          -0.00043021927331693947,
          0.3846197472407903
        ],
        "mug_pos": [
          0.49159437735673317,
          -0.00014800444123032555,
          0.37979619065988274
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0650887540655682e-05,
        -0.058692909828880785,
        -2.1212664279828964e-06,
        -1.868802839730346,
        3.139509619849332e-05,
        1.604786021562814,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -9.732915721441689e-06,
          -0.025529950756124995,
          8.23387475466534e-06,
          -1.9006513973900592,
          -0.002071501972857337,
          1.59716947946067,
          -0.7852966118717185
        ],
        "ee_pos": [
          0.4932815134186604,
          -0.0004253364468671861,
          0.3910506327262663
        ],
        "mug_pos": [
          0.4919739548871683,
          -0.00014321300041688763,
          0.3860306507483147
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.066307672273607e-05,
        -0.06840902326695092,
        -2.140003322276954e-06,
        -1.8635799013309382,
        3.145618971307847e-05,
        1.603237428466883,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -8.563040306403928e-06,
          -0.034597539284886555,
          2.1308458508553384e-05,
          -1.8959778123759696,
          -0.00207263293265324,
          1.595942456932944,
          -0.7852967197041696
        ],
        "ee_pos": [
          0.49364545257234543,
          -0.0004192029879830212,
          0.3974628949764172
        ],
        "mug_pos": [
          0.4923604795746099,
          -0.0001371439640710754,
          0.39225251402183514
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0671871923238932e-05,
        -0.07595733360824007,
        -2.1545965737234786e-06,
        -1.859265993667638,
        3.1508076197480736e-05,
        1.6022305561543206,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -6.964271831262252e-06,
          -0.043306852083406476,
          3.479710354832299e-05,
          -1.891274500945989,
          -0.002061278546982814,
          1.5950707599014395,
          -0.785297047376843
        ],
        "ee_pos": [
          0.4940404236356721,
          -0.00040946589461966895,
          0.40373809725278514
        ],
        "mug_pos": [
          0.4927809086067395,
          -0.00012733681245955704,
          0.3983419043928701
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0680787461258631e-05,
        -0.0839002094175401,
        -2.1702795571559494e-06,
        -1.8545420451739878,
        3.156576055114477e-05,
        1.60131529261912,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -5.012358330301694e-06,
          -0.05171078744236261,
          4.861576441739685e-05,
          -1.8865553231933008,
          -0.0020714635307502493,
          1.5942690492595788,
          -0.785297583192563
        ],
        "ee_pos": [
          0.4944006516256011,
          -0.00040351179451615844,
          0.4098852478488483
        ],
        "mug_pos": [
          0.4931677481942001,
          -0.00012140735482427747,
          0.4043072535831798
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0689440543219069e-05,
        -0.09178918059894553,
        -2.1863402197006297e-06,
        -1.849687323428198,
        3.1625711948834694e-05,
        1.6005356803474398,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -3.867176043542187e-06,
          -0.05986471182085581,
          6.110099744331618e-05,
          -1.8818185183605178,
          -0.0020920326990802865,
          1.5934840363605904,
          -0.7852976461552896
        ],
        "ee_pos": [
          0.49471064541587667,
          -0.00040094232193778947,
          0.41593018987887753
        ],
        "mug_pos": [
          0.4935052074907904,
          -0.0001189543198187107,
          0.4101736904713451
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.069763190218879e-05,
        -0.09945322024889207,
        -2.2023857509695263e-06,
        -1.844803166395095,
        3.1686713369244936e-05,
        1.5999121455462602,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -3.938814970064686e-06,
          -0.06778279778865764,
          7.157397731631326e-05,
          -1.8770587881701426,
          -0.0021026368258704404,
          1.5928573667810608,
          -0.7852974631874355
        ],
        "ee_pos": [
          0.4950045765937593,
          -0.0003980372940455869,
          0.4218882022094911
        ],
        "mug_pos": [
          0.4938285688211975,
          -0.00011612984649000596,
          0.4159560937679952
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0705269882196186e-05,
        -0.10681788117785862,
        -2.2181829527788487e-06,
        -1.8399385525601581,
        3.174817111495223e-05,
        1.5994497559732799,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.5279246320497554e-06,
          -0.07544953635717366,
          8.094108881953613e-05,
          -1.8722758599674503,
          -0.0021035991576794978,
          1.5924685387268036,
          -0.785297607161314
        ],
        "ee_pos": [
          0.49530673626453375,
          -0.0003938563811344153,
          0.4277568343823218
        ],
        "mug_pos": [
          0.4941630684069076,
          -0.00011194642682994915,
          0.4216523061747643
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0712319850693066e-05,
        -0.11384411070356512,
        -2.2335765733542573e-06,
        -1.8351279108311513,
        3.180963282160257e-05,
        1.599144394353308,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.866135142307855e-06,
          -0.08283667640899013,
          9.027245086753041e-05,
          -1.8674886804018247,
          -0.0021035120496960104,
          1.5922755651584795,
          -0.7852980115349313
        ],
        "ee_pos": [
          0.49561135689539176,
          -0.0003892457801841257,
          0.433513928345126
        ],
        "mug_pos": [
          0.49450220707140435,
          -0.00010731888568879968,
          0.42724077728466375
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0718776543488957e-05,
        -0.12050722568609451,
        -2.2484504785448123e-06,
        -1.8304020627384945,
        3.187066103159148e-05,
        1.5989863723118491,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.690541083501314e-06,
          -0.08991021091699528,
          9.993671918894665e-05,
          -1.8627363831906851,
          -0.002104492816404348,
          1.5922104363635132,
          -0.7852983098772224
        ],
        "ee_pos": [
          0.49590269486683913,
          -0.00038439977408535517,
          0.4391230539625342
        ],
        "mug_pos": [
          0.4948292891462852,
          -0.00010245421384119869,
          0.43268535276288217
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0724650018281687e-05,
        -0.12678948724954955,
        -2.2627101165574203e-06,
        -1.825791084195201,
        3.193080733244089e-05,
        1.5989624500108304,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -4.033045245065384e-06,
          -0.09663359755058706,
          0.00010986653578381568,
          -1.8580664917377419,
          -0.0021045492298903917,
          1.592244732384924,
          -0.7852984909755771
        ],
        "ee_pos": [
          0.496173294551146,
          -0.00037896155523504667,
          0.44454317197885296
        ],
        "mug_pos": [
          0.4951360897199512,
          -9.69718916473207e-05,
          0.4379449368188542
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0729958645869982e-05,
        -0.13267733964004946,
        -2.2762749029738265e-06,
        -1.821324817496582,
        3.1989612801617144e-05,
        1.5990570093500327,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -2.9798477374883536e-06,
          -0.10297269066296828,
          0.00011991955179407108,
          -1.8535239159016315,
          -0.002103656431319883,
          1.5923758239449652,
          -0.785298683188755
        ],
        "ee_pos": [
          0.4964236358200685,
          -0.00037300339428694663,
          0.44973600141389525
        ],
        "mug_pos": [
          0.4954223165913667,
          -9.093931390231615e-05,
          0.4429811232837
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0734725520769694e-05,
        -0.1381602699727139,
        -2.289075006341412e-06,
        -1.817032742367632,
        3.204661420875629e-05,
        1.5992528097141063,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.6726364969083915e-06,
          -0.10890054467284274,
          0.00012988805518487852,
          -1.8491474464132955,
          -0.0021034474095026063,
          1.5925947637860807,
          -0.785298901453153
        ],
        "ee_pos": [
          0.49665482741506195,
          -0.00036703068687211036,
          0.4546688153907743
        ],
        "mug_pos": [
          0.4956884170349794,
          -8.487400308614544e-05,
          0.44776107075365573
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0738976487233258e-05,
        -0.14323023669729962,
        -2.3010499798969916e-06,
        -1.8129436966425916,
        3.210135081698424e-05,
        1.5995315462642254,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          2.381127426480379e-07,
          -0.11427041569978863,
          0.0001395895594149973,
          -1.8450864279287293,
          -0.002083045752894827,
          1.592937396227047,
          -0.7852992139248158
        ],
        "ee_pos": [
          0.496871785270518,
          -0.00035704797727927594,
          0.4592137393537428
        ],
        "mug_pos": [
          0.49593835231397554,
          -7.460719213081453e-05,
          0.4521597335614567
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.074183477107311e-05,
        -0.14686613992774558,
        -2.3096156141537674e-06,
        -1.8099188440157827,
        3.2142209867728486e-05,
        1.5998054797536816,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          2.5527322957115217e-06,
          -0.11914251015834047,
          0.0001503323189980366,
          -1.8412790184655072,
          -0.002096751313877683,
          1.5932065597525467,
          -0.7852995124245107
        ],
        "ee_pos": [
          0.4970490600293268,
          -0.00035231992366492485,
          0.46338723945205423
        ],
        "mug_pos": [
          0.4961431832941633,
          -6.979987884532564e-05,
          0.4561880310993376
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0745681743911637e-05,
        -0.15178078189585428,
        -2.321411978523858e-06,
        -1.8057466069629675,
        3.21987994169059e-05,
        1.6002452046972535,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          3.7879982448816876e-06,
          -0.12356759407485281,
          0.00015862056551705622,
          -1.8377578283709985,
          -0.002109311959355949,
          1.5934913710560352,
          -0.7852993762338858
        ],
        "ee_pos": [
          0.4971980230761397,
          -0.0003500803000220597,
          0.4672260555986607
        ],
        "mug_pos": [
          0.49631700067714163,
          -6.751357307054337e-05,
          0.4598857232003544
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0748044142849736e-05,
        -0.15492949873599718,
        -2.329006901162724e-06,
        -1.8030064642695494,
        3.2236198514847445e-05,
        1.600581625652046,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          4.142029207768874e-06,
          -0.12749061349880533,
          0.00016504974598150117,
          -1.834570162319252,
          -0.0021042002684569807,
          1.5938209468476436,
          -0.7852993659883276
        ],
        "ee_pos": [
          0.4973343843032385,
          -0.0003453479164399105,
          0.4706716112397659
        ],
        "mug_pos": [
          0.496473654787119,
          -6.259190004671332e-05,
          0.463192986779078
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0750348004014183e-05,
        -0.15809411893920536,
        -2.3366774013643416e-06,
        -1.8001989042620279,
        3.2274695777844874e-05,
        1.6009635268570768,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          4.1895986094206976e-06,
          -0.13096408813957858,
          0.00017066941596205972,
          -1.8316800518963505,
          -0.0021083697491158584,
          1.5941294391536345,
          -0.7852995493199647
        ],
        "ee_pos": [
          0.4974514591477347,
          -0.0003428613674673254,
          0.47375844764682845
        ],
        "mug_pos": [
          0.4966060848222308,
          -5.997069096589575e-05,
          0.46614290533427766
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0752453069283167e-05,
        -0.16103851551906614,
        -2.343875469622987e-06,
        -1.797540556511,
        3.231128564431875e-05,
        1.6013569149233386,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          3.6260600480708156e-06,
          -0.13405334731939225,
          0.00017504114202932048,
          -1.829056015983372,
          -0.0021218984857895645,
          1.5944149046068186,
          -0.785299540188991
        ],
        "ee_pos": [
          0.49755115121971605,
          -0.00034336225221883506,
          0.476532567887215
        ],
        "mug_pos": [
          0.4967166727240322,
          -6.0450592134356364e-05,
          0.46878091672823247
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0754290897436263e-05,
        -0.16365216016068754,
        -2.3503157344947294e-06,
        -1.795141727159823,
        3.234441866432361e-05,
        1.6017383738925917,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          2.7933017247847514e-06,
          -0.13669294238435106,
          0.00017774313016781496,
          -1.8268138552986217,
          -0.0021096798310147294,
          1.594704226011721,
          -0.7852995727445758
        ],
        "ee_pos": [
          0.4976343718418963,
          -0.000340072730344389,
          0.4789220869755104
        ],
        "mug_pos": [
          0.4968276328513857,
          -5.6932595161563685e-05,
          0.47103388281243086
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0755193430075252e-05,
        -0.16497990431320278,
        -2.353584102251102e-06,
        -1.7939072374229967,
        3.236152471858211e-05,
        1.6019451409424792,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          2.94962342927111e-06,
          -0.13881029560753597,
          0.00018134363085540293,
          -1.825000102462074,
          -0.0020956665471438814,
          1.5949103512753862,
          -0.7852998416944573
        ],
        "ee_pos": [
          0.4976909227746672,
          -0.0003349771516070251,
          0.48085081507587335
        ],
        "mug_pos": [
          0.4969033254727104,
          -5.152662574470715e-05,
          0.47282745285018435
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0756111015564458e-05,
        -0.16635583312394084,
        -2.35697279482858e-06,
        -1.792616857235379,
        3.237944153984144e-05,
        1.6021684991839913,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          3.7123513298150238e-06,
          -0.14046720405446597,
          0.00018579244183747408,
          -1.8235589103573762,
          -0.002108333890845714,
          1.5950005482205862,
          -0.7852999089232819
        ],
        "ee_pos": [
          0.49771838461673523,
          -0.00033448035956072837,
          0.48236727446852307
        ],
        "mug_pos": [
          0.4969449417296719,
          -5.093488740387783e-05,
          0.4742087822299302
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0756956499714357e-05,
        -0.16762906491674973,
        -2.360124145050482e-06,
        -1.7914128409266996,
        3.239618594903106e-05,
        1.6023833955239783,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          3.5874624935361432e-06,
          -0.14176113019208247,
          0.0001888461361494126,
          -1.8224152894040577,
          -0.002132060069978517,
          1.595056858864368,
          -0.7852996082787014
        ],
        "ee_pos": [
          0.49773570838435743,
          -0.00033764735505794285,
          0.4835589680025074
        ],
        "mug_pos": [
          0.4969716830481106,
          -5.415543990186642e-05,
          0.4752650934028536
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          2.465121848415572e-06,
          -0.14268189176364485,
          0.0001893046999160625,
          -1.821625027099815,
          -0.002127043640751199,
          1.5951341428718693,
          -0.7852994940102409
        ],
        "ee_pos": [
          0.49775092940175447,
          -0.0003372188366809693,
          0.48441090095425593
        ],
        "mug_pos": [
          0.49698972429787247,
          -5.356890184147169e-05,
          0.47598281068014997
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          1.744942783840882e-06,
          -0.14318932449910216,
          0.00018972999514760733,
          -1.8212034881064216,
          -0.002114296709033042,
          1.5951339898607557,
          -0.7852997541090441
        ],
        "ee_pos": [
          0.4977483537275905,
          -0.00033461332790977484,
          0.48488058617839774
        ],
        "mug_pos": [
          0.4969839322105991,
          -5.068676218424639e-05,
          0.4763189770202377
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          1.5062731181137574e-06,
          -0.1433470043921933,
          0.00019079948817237033,
          -1.821081952437128,
          -0.00212470993224157,
          1.595031749378308,
          -0.7852997918711484
        ],
        "ee_pos": [
          0.49772435205285903,
          -0.0003360170094107252,
          0.48502498332806004
        ],
        "mug_pos": [
          0.4969534193111723,
          -5.200559235047414e-05,
          0.4763295283011684
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          5.349557880220025e-07,
          -0.14326520767061499,
          0.0001908466373701447,
          -1.8211606964015778,
          -0.00214979317988061,
          1.5949070517221207,
          -0.7852994212856618
        ],
        "ee_pos": [
          0.4976961518493438,
          -0.0003414960475851673,
          0.48494913039174536
        ],
        "mug_pos": [
          0.4969135756829439,
          -5.759276910586831e-05,
          0.4761193147079345
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.8400951883031678e-06,
          -0.14305879695576162,
          0.0001886759490269226,
          -1.8213366512444378,
          -0.0021699033242825443,
          1.5948240442146633,
          -0.7852990420518008
        ],
        "ee_pos": [
          0.49767856545172356,
          -0.00034795928847645844,
          0.4847630941136192
        ],
        "mug_pos": [
          0.49687944040786974,
          -6.419019966701208e-05,
          0.47579847755752663
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -5.181282742841004e-06,
          -0.1428230914906447,
          0.00018474335912709215,
          -1.8215281023501029,
          -0.0021796116628276484,
          1.5947859556064141,
          -0.78529888168195
        ],
        "ee_pos": [
          0.497672581519255,
          -0.0003537072877617528,
          0.4845541387655064
        ],
        "mug_pos": [
          0.496858730518931,
          -7.001871745605914e-05,
          0.4754539666154622
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -8.613845096239851e-06,
          -0.1426212373949243,
          0.00018031074797029073,
          -1.8216857460360825,
          -0.0021834250535852527,
          1.5947735213167251,
          -0.785298832626629
        ],
        "ee_pos": [
          0.4976729075870279,
          -0.0003585447581986216,
          0.4843770348655235
        ],
        "mug_pos": [
          0.49684474102926685,
          -7.489957876907202e-05,
          0.4751409294725056
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.1457210787897908e-05,
          -0.14248307214075692,
          0.00017645176936009977,
          -1.8217898700205342,
          -0.002184716287976203,
          1.5947724642358465,
          -0.7852987874184879
        ],
        "ee_pos": [
          0.4976754769956352,
          -0.0003622953942502079,
          0.4842564197465013
        ],
        "mug_pos": [
          0.49683329170889584,
          -7.868025632334639e-05,
          0.47488422238024297
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.3354902916949661e-05,
          -0.14241173705427337,
          0.000173772002184101,
          -1.821840778444861,
          -0.0021837652142539473,
          1.5947801701660953,
          -0.7852987731425681
        ],
        "ee_pos": [
          0.49767903959674314,
          -0.00036452770687494753,
          0.48419459752356797
        ],
        "mug_pos": [
          0.4968239857138834,
          -8.09245732635253e-05,
          0.47468623537419546
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.4207388322302199e-05,
          -0.14239435808210177,
          0.00017249177949911425,
          -1.8218496927500587,
          -0.0021808926316301317,
          1.5947946407917384,
          -0.7852988100690144
        ],
        "ee_pos": [
          0.4976832315348576,
          -0.00036511303127385473,
          0.4841804215924188
        ],
        "mug_pos": [
          0.49681653059533404,
          -8.150393176473362e-05,
          0.47453590309223637
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.4130106787577464e-05,
          -0.14241195884635732,
          0.00017250998618028893,
          -1.8218320329049218,
          -0.00217747439439771,
          1.5948103724715688,
          -0.7852988810008571
        ],
        "ee_pos": [
          0.49768701202779614,
          -0.00036442201751167797,
          0.4841970615120463
        ],
        "mug_pos": [
          0.4968097381962909,
          -8.079466767026064e-05,
          0.4744164398640385
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.3407138547534001e-05,
          -0.1424461203011454,
          0.00017346469038480108,
          -1.8218026516391417,
          -0.0021747337665007772,
          1.5948220508707571,
          -0.7852989431069137
        ],
        "ee_pos": [
          0.4976895230378475,
          -0.00036303140067796486,
          0.4842279995550629
        ],
        "mug_pos": [
          0.4968013208322475,
          -7.938906269577532e-05,
          0.47431143129418524
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.2389430412440504e-05,
          -0.14248257077167029,
          0.000174871469568329,
          -1.821772849954203,
          -0.00217303231376666,
          1.5948283656957831,
          -0.7852989834743048
        ],
        "ee_pos": [
          0.49769063259930263,
          -0.00036144853668586394,
          0.48426053380284906
        ],
        "mug_pos": [
          0.49678994686332023,
          -7.780040701342182e-05,
          0.47420824467940204
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.1386012483379138e-05,
          -0.142512514447013,
          0.0001762870220704812,
          -1.82174933229013,
          -0.0021722140342319753,
          1.594830118674492,
          -0.7852990060793738
        ],
        "ee_pos": [
          0.4976906053176894,
          -0.00036003341719894835,
          0.4842869306975859
        ],
        "mug_pos": [
          0.4967796212524996,
          -7.637144756407629e-05,
          0.47409877748674367
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0599660101555188e-05,
          -0.14253216573081154,
          0.00017741346375052427,
          -1.8217345233055733,
          -0.0021720609352510983,
          1.594829753749088,
          -0.7852990139633721
        ],
        "ee_pos": [
          0.4976901641280012,
          -0.00035900243383251926,
          0.4843041141051488
        ],
        "mug_pos": [
          0.49676872639188885,
          -7.533016586580574e-05,
          0.47398012675940615
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0118445739245685e-05,
          -0.14254153428118466,
          0.00017811400198642963,
          -1.8217278713266098,
          -0.0021723747114253604,
          1.5948285260363253,
          -0.7852990106405788
        ],
        "ee_pos": [
          0.49768960734376155,
          -0.00035843854861836687,
          0.4843122585610268
        ],
        "mug_pos": [
          0.4967566975016052,
          -7.475567286976887e-05,
          0.47385251637427006
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -9.938793556417337e-06,
          -0.14254299061149656,
          0.00017838626223419806,
          -1.8217274161375823,
          -0.0021729219869614878,
          1.5948262526523034,
          -0.7852990004364389
        ],
        "ee_pos": [
          0.4976889804218299,
          -0.0003582988246620686,
          0.4843133827413415
        ],
        "mug_pos": [
          0.49674353575708236,
          -7.46002493762025e-05,
          0.47371796553393786
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -9.993092462695816e-06,
          -0.14253951313058696,
          0.00017832051088262197,
          -1.8217306513161464,
          -0.002173463326784206,
          1.5948240873611608,
          -0.78529898074578
        ],
        "ee_pos": [
          0.4976885248894231,
          -0.0003584567954168823,
          0.4843102048776379
        ],
        "mug_pos": [
          0.4967285278735255,
          -7.475802710528182e-05,
          0.4735792702569178
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0187043913187139e-05,
          -0.14253400626696172,
          0.00017804929442156173,
          -1.8217353832170897,
          -0.0021738970441424585,
          1.5948220074116615,
          -0.7852989759775195
        ],
        "ee_pos": [
          0.4976879317852255,
          -0.0003587782101001502,
          0.48430514165142546
        ],
        "mug_pos": [
          0.49671682222507274,
          -7.508620061455865e-05,
          0.47343837795544585
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0429911118781978e-05,
          -0.14252845141581066,
          0.00017770440176602742,
          -1.8217399327250066,
          -0.00217416816540967,
          1.594821213876545,
          -0.7852989699804196
        ],
        "ee_pos": [
          0.4976879471081606,
          -0.00035913189224882533,
          0.4843002173155555
        ],
        "mug_pos": [
          0.49670435195342194,
          -7.543820751501777e-05,
          0.47329754965613785
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0651605389588357e-05,
          -0.14252405943138838,
          0.00017738638355057016,
          -1.821743350792104,
          -0.002174293935502057,
          1.5948212623537732,
          -0.785298960381463
        ],
        "ee_pos": [
          0.49768794506151043,
          -0.00035943733855680476,
          0.48429635026307016
        ],
        "mug_pos": [
          0.49669416815749934,
          -7.574849802338469e-05,
          0.473157753655232
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0813095392201882e-05,
          -0.14252127763734151,
          0.00017715246001866222,
          -1.821745308372684,
          -0.0021743078221389935,
          1.5948220987601502,
          -0.7852989594708206
        ],
        "ee_pos": [
          0.4976881431909126,
          -0.00035964784726313427,
          0.4842940126521116
        ],
        "mug_pos": [
          0.49668255435320396,
          -7.595922327979791e-05,
          0.4730194654364245
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0903877786367561e-05,
          -0.1425201286961705,
          0.00017702088346637195,
          -1.821746083268209,
          -0.0021742471504536624,
          1.5948220901672934,
          -0.7852989653933475
        ],
        "ee_pos": [
          0.49768825713216475,
          -0.00035975392175638876,
          0.4842930274392983
        ],
        "mug_pos": [
          0.49667054315651316,
          -7.605019548759861e-05,
          0.4728824792236475
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0929657723631194e-05,
          -0.14252015321996558,
          0.00017698194879479888,
          -1.821745984893298,
          -0.002174141590620326,
          1.5948221079570575,
          -0.7852989631340114
        ],
        "ee_pos": [
          0.497688283683839,
          -0.0003597696263846499,
          0.484293039068646
        ],
        "mug_pos": [
          0.49665983134640107,
          -7.605976789596368e-05,
          0.4727464728848256
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0908734561650548e-05,
          -0.14252080422047858,
          0.00017701044123827266,
          -1.821745368890625,
          -0.002174047247864409,
          1.5948228843230061,
          -0.7852989680480261
        ],
        "ee_pos": [
          0.4976884179336335,
          -0.0003597279563233235,
          0.48429366833384563
        ],
        "mug_pos": [
          0.49664839077943757,
          -7.599787556023488e-05,
          0.472611081806209
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0863601882040712e-05,
          -0.14252171924505116,
          0.0001770750254774268,
          -1.8217445661763783,
          -0.00217398461235588,
          1.5948234189035353,
          -0.7852989669216014
        ],
        "ee_pos": [
          0.4976885006008943,
          -0.00035966044296606464,
          0.48429451743880025
        ],
        "mug_pos": [
          0.4966372773279021,
          -7.590820681039106e-05,
          0.4724758675930898
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0812888915018629e-05,
          -0.14252254991959032,
          0.00017714778071927458,
          -1.8217438117694291,
          -0.0021739342298849754,
          1.594824153097048,
          -0.7852989630983787
        ],
        "ee_pos": [
          0.49768864938870194,
          -0.0003595868750339375,
          0.4842953354392928
        ],
        "mug_pos": [
          0.49662483515859634,
          -7.582350608888726e-05,
          0.47234092229332064
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0770816219603616e-05,
          -0.14252330188191206,
          0.00017720994455085328,
          -1.8217433074855793,
          -0.0021739355753142096,
          1.5948233411322166,
          -0.785298973397457
        ],
        "ee_pos": [
          0.49768862147596843,
          -0.0003595315892268349,
          0.48429594411959365
        ],
        "mug_pos": [
          0.4966130700600731,
          -7.574263832753873e-05,
          0.4722054358662229
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0741583783673928e-05,
          -0.14252369488505545,
          0.00017725119599339557,
          -1.8217429837512478,
          -0.0021739228531994104,
          1.5948237191901524,
          -0.7852989679658585
        ],
        "ee_pos": [
          0.4976885227528972,
          -0.00035949379224805027,
          0.4842962848085677
        ],
        "mug_pos": [
          0.49660284856762826,
          -7.567832812990236e-05,
          0.4720696764793831
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.07274749228242e-05,
          -0.142523872021005,
          0.00017727247955616132,
          -1.821742858052072,
          -0.002173937271796389,
          1.5948238041614329,
          -0.7852989676880122
        ],
        "ee_pos": [
          0.4976886734505064,
          -0.0003594759054509848,
          0.48429648328174063
        ],
        "mug_pos": [
          0.4965907308241048,
          -7.562723248874966e-05,
          0.47193382653212657
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 0.0,
      "is_first": false,
      "is_last": false,
      "is_terminal": false,
      "language_instruction": "pick up the mug"
    },
    {
      "observation": {
        "qpos_arm": [
          -1.0724689831325491e-05,
          -0.14252386379251114,
          0.00017727619861632911,
          -1.821742829234499,
          -0.0021739582001233768,
          1.594824014297118,
          -0.7852989664777631
        ],
        "ee_pos": [
          0.49768859328000886,
          -0.00035947751023727894,
          0.4842964798276259
        ],
        "mug_pos": [
          0.496579732121717,
          -7.559812230924723e-05,
          0.4717977774117769
        ],
        "gripper_state": 0.0
      },
      "action": [
        -1.0757668872601197e-05,
        -0.16869706370424517,
        -2.362785870337109e-06,
        -1.79039485730932,
        3.2410362316095674e-05,
        1.6025702792382515,
        -0.7853,
        0.0
      ],
      "reward": 1.0,
      "is_first": false,
      "is_last": true,
      "is_terminal": true,
      "language_instruction": "pick up the mug"
    }
  ]
}