openadapt-ml 0.2.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (95) hide show
  1. openadapt_ml/baselines/__init__.py +121 -0
  2. openadapt_ml/baselines/adapter.py +185 -0
  3. openadapt_ml/baselines/cli.py +314 -0
  4. openadapt_ml/baselines/config.py +448 -0
  5. openadapt_ml/baselines/parser.py +922 -0
  6. openadapt_ml/baselines/prompts.py +787 -0
  7. openadapt_ml/benchmarks/__init__.py +13 -115
  8. openadapt_ml/benchmarks/agent.py +265 -421
  9. openadapt_ml/benchmarks/azure.py +28 -19
  10. openadapt_ml/benchmarks/azure_ops_tracker.py +521 -0
  11. openadapt_ml/benchmarks/cli.py +1722 -4847
  12. openadapt_ml/benchmarks/trace_export.py +631 -0
  13. openadapt_ml/benchmarks/viewer.py +22 -5
  14. openadapt_ml/benchmarks/vm_monitor.py +530 -29
  15. openadapt_ml/benchmarks/waa_deploy/Dockerfile +47 -53
  16. openadapt_ml/benchmarks/waa_deploy/api_agent.py +21 -20
  17. openadapt_ml/cloud/azure_inference.py +3 -5
  18. openadapt_ml/cloud/lambda_labs.py +722 -307
  19. openadapt_ml/cloud/local.py +2038 -487
  20. openadapt_ml/cloud/ssh_tunnel.py +68 -26
  21. openadapt_ml/datasets/next_action.py +40 -30
  22. openadapt_ml/evals/grounding.py +8 -3
  23. openadapt_ml/evals/plot_eval_metrics.py +15 -13
  24. openadapt_ml/evals/trajectory_matching.py +41 -26
  25. openadapt_ml/experiments/demo_prompt/format_demo.py +16 -6
  26. openadapt_ml/experiments/demo_prompt/run_experiment.py +26 -16
  27. openadapt_ml/experiments/representation_shootout/__init__.py +70 -0
  28. openadapt_ml/experiments/representation_shootout/conditions.py +708 -0
  29. openadapt_ml/experiments/representation_shootout/config.py +390 -0
  30. openadapt_ml/experiments/representation_shootout/evaluator.py +659 -0
  31. openadapt_ml/experiments/representation_shootout/runner.py +687 -0
  32. openadapt_ml/experiments/waa_demo/runner.py +29 -14
  33. openadapt_ml/export/parquet.py +36 -24
  34. openadapt_ml/grounding/detector.py +18 -14
  35. openadapt_ml/ingest/__init__.py +8 -6
  36. openadapt_ml/ingest/capture.py +25 -22
  37. openadapt_ml/ingest/loader.py +7 -4
  38. openadapt_ml/ingest/synthetic.py +189 -100
  39. openadapt_ml/models/api_adapter.py +14 -4
  40. openadapt_ml/models/base_adapter.py +10 -2
  41. openadapt_ml/models/providers/__init__.py +288 -0
  42. openadapt_ml/models/providers/anthropic.py +266 -0
  43. openadapt_ml/models/providers/base.py +299 -0
  44. openadapt_ml/models/providers/google.py +376 -0
  45. openadapt_ml/models/providers/openai.py +342 -0
  46. openadapt_ml/models/qwen_vl.py +46 -19
  47. openadapt_ml/perception/__init__.py +35 -0
  48. openadapt_ml/perception/integration.py +399 -0
  49. openadapt_ml/retrieval/demo_retriever.py +50 -24
  50. openadapt_ml/retrieval/embeddings.py +9 -8
  51. openadapt_ml/retrieval/retriever.py +3 -1
  52. openadapt_ml/runtime/__init__.py +50 -0
  53. openadapt_ml/runtime/policy.py +18 -5
  54. openadapt_ml/runtime/safety_gate.py +471 -0
  55. openadapt_ml/schema/__init__.py +9 -0
  56. openadapt_ml/schema/converters.py +74 -27
  57. openadapt_ml/schema/episode.py +31 -18
  58. openadapt_ml/scripts/capture_screenshots.py +530 -0
  59. openadapt_ml/scripts/compare.py +85 -54
  60. openadapt_ml/scripts/demo_policy.py +4 -1
  61. openadapt_ml/scripts/eval_policy.py +15 -9
  62. openadapt_ml/scripts/make_gif.py +1 -1
  63. openadapt_ml/scripts/prepare_synthetic.py +3 -1
  64. openadapt_ml/scripts/train.py +21 -9
  65. openadapt_ml/segmentation/README.md +920 -0
  66. openadapt_ml/segmentation/__init__.py +97 -0
  67. openadapt_ml/segmentation/adapters/__init__.py +5 -0
  68. openadapt_ml/segmentation/adapters/capture_adapter.py +420 -0
  69. openadapt_ml/segmentation/annotator.py +610 -0
  70. openadapt_ml/segmentation/cache.py +290 -0
  71. openadapt_ml/segmentation/cli.py +674 -0
  72. openadapt_ml/segmentation/deduplicator.py +656 -0
  73. openadapt_ml/segmentation/frame_describer.py +788 -0
  74. openadapt_ml/segmentation/pipeline.py +340 -0
  75. openadapt_ml/segmentation/schemas.py +622 -0
  76. openadapt_ml/segmentation/segment_extractor.py +634 -0
  77. openadapt_ml/training/azure_ops_viewer.py +1097 -0
  78. openadapt_ml/training/benchmark_viewer.py +52 -41
  79. openadapt_ml/training/shared_ui.py +7 -7
  80. openadapt_ml/training/stub_provider.py +57 -35
  81. openadapt_ml/training/trainer.py +143 -86
  82. openadapt_ml/training/trl_trainer.py +70 -21
  83. openadapt_ml/training/viewer.py +323 -108
  84. openadapt_ml/training/viewer_components.py +180 -0
  85. {openadapt_ml-0.2.0.dist-info → openadapt_ml-0.2.1.dist-info}/METADATA +215 -14
  86. openadapt_ml-0.2.1.dist-info/RECORD +116 -0
  87. openadapt_ml/benchmarks/base.py +0 -366
  88. openadapt_ml/benchmarks/data_collection.py +0 -432
  89. openadapt_ml/benchmarks/live_tracker.py +0 -180
  90. openadapt_ml/benchmarks/runner.py +0 -418
  91. openadapt_ml/benchmarks/waa.py +0 -761
  92. openadapt_ml/benchmarks/waa_live.py +0 -619
  93. openadapt_ml-0.2.0.dist-info/RECORD +0 -86
  94. {openadapt_ml-0.2.0.dist-info → openadapt_ml-0.2.1.dist-info}/WHEEL +0 -0
  95. {openadapt_ml-0.2.0.dist-info → openadapt_ml-0.2.1.dist-info}/licenses/LICENSE +0 -0
openadapt_ml-0.2.0.dist-info/RECORD DELETED
@@ -1,86 +0,0 @@
1
- openadapt_ml/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
2
- openadapt_ml/config.py,sha256=eH5WTKRPkkidjkNb25Wn_dUAizUQTsVPMYLDq_ekPJQ,1865
3
- openadapt_ml/benchmarks/__init__.py,sha256=RRAoXm_al-DgStCKCycnC60_iZo52mLmUu5nN5IPfxY,3855
4
- openadapt_ml/benchmarks/agent.py,sha256=kuq-dWqWJH8ogjklAFdnbG_6zVzHtd6Ab2rPxRjvHtU,29724
5
- openadapt_ml/benchmarks/azure.py,sha256=AI5ZdsNxmQ197ymOj68lTLAtl-eByuguhACqv7UZhIg,28145
6
- openadapt_ml/benchmarks/base.py,sha256=d7T_zMlMPlN0beDWkpzOvOHYQO6QnsePLQ45iKbi66Y,11667
7
- openadapt_ml/benchmarks/cli.py,sha256=ElBZDcmDOpA0uAoHYKRx3fLAJ5v8dH4R9SxfhmwTZWw,204898
8
- openadapt_ml/benchmarks/data_collection.py,sha256=EYOsYnFQifF3MXD0TZxznd-HbODiovnGDtxGjSMpO-Y,14652
9
- openadapt_ml/benchmarks/live_tracker.py,sha256=1SukwgRYbAzSMzHOhPZOSgZ58L44CYtX-KOfAzyAJZw,5130
10
- openadapt_ml/benchmarks/runner.py,sha256=a52GasPKEnNgevxilQAI1z8FnzwWddLiDY60NYWxpZk,13616
11
- openadapt_ml/benchmarks/viewer.py,sha256=Mjbt67gfnSw29rNhU4JYL0cdAc-UZ3QpxmG6GhKnD68,41111
12
- openadapt_ml/benchmarks/vm_monitor.py,sha256=NKhLF66hcBqxIDHk-iPKn31AfzpXyE-9IkBTYX2k70I,20330
13
- openadapt_ml/benchmarks/waa.py,sha256=V_RtDfOLW2jD8MN-M9YJYvt67NQCXrUkVxOtcNYVLHo,27008
14
- openadapt_ml/benchmarks/waa_live.py,sha256=U92KaBWCpJmwqSURJeG5biEolTyBCvyCikWRz05IV-A,22982
15
- openadapt_ml/benchmarks/waa_deploy/Dockerfile,sha256=m9Nsei1v_m0jF1HWc7yCMz-_wD3rHHv9H9HBC2lFVSE,10969
16
- openadapt_ml/benchmarks/waa_deploy/__init__.py,sha256=KV71HrrgETytfY0i4vFSi-yM0KjoQP2hd9Bl03cZ9yc,320
17
- openadapt_ml/benchmarks/waa_deploy/api_agent.py,sha256=dbazhRxc554901LFiVuj6sMmOgoHtTKl8XIAiIJrFWU,20024
18
- openadapt_ml/benchmarks/waa_deploy/start_waa_server.bat,sha256=YxgrSWh76zLijlpxEpulWf9To2JtJ-yR42lr2WyTXiY,1496
19
- openadapt_ml/cloud/__init__.py,sha256=XYrvxivJeZ8qYnuGod5kodMlm3iT2OK2GAApO3CNB0c,133
20
- openadapt_ml/cloud/azure_inference.py,sha256=lqkG86Dn6x2Rp7Y-X9tK_8mn0QfRvaDQNRHg-kJkQyc,15736
21
- openadapt_ml/cloud/lambda_labs.py,sha256=jZ700gNR8zxO2qGa7yqByKXgb4-18zN30YQdDZ3Fbuk,102206
22
- openadapt_ml/cloud/local.py,sha256=AX7gxgkwGwOJ1ogp-rQefKDyK3MDNMWA9W2qkwaV66g,104782
23
- openadapt_ml/cloud/ssh_tunnel.py,sha256=q4VEKT4cEU0-hAP1t9fL2jRur0FMRxg8ZV6j1IQV45k,20824
24
- openadapt_ml/datasets/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
25
- openadapt_ml/datasets/next_action.py,sha256=eNNwwdRE-pJa1DCgEIr9eTipZ7pDAEDVxkc9UPObJLs,21219
26
- openadapt_ml/evals/__init__.py,sha256=Kx7bSvPHwmoGVI3q1wS_lC17W2S32YHj0459JRqu6Ow,573
27
- openadapt_ml/evals/grounding.py,sha256=un4bWrhqwrJ4O9QqF40NfnhKeOMntGdCJcwXz6ZzNJ4,8447
28
- openadapt_ml/evals/plot_eval_metrics.py,sha256=gsikQ3MSUY7Pw61D8lB286q0MPpBL9E05UFHPrawViw,5237
29
- openadapt_ml/evals/trajectory_matching.py,sha256=p40wDDlD0AyiY6vsgkcp6FBDISyKKQKycsrEz7uGF8Y,20616
30
- openadapt_ml/experiments/demo_prompt/__init__.py,sha256=dwS0bI53jXMzHE-DPhb_mhmPdoqSZRIcNbV79wt8KPM,454
31
- openadapt_ml/experiments/demo_prompt/format_demo.py,sha256=rIXcjiYhI2YuLDUjY4iTxPp4ZmtvKibQQYTCfEn4lZs,6553
32
- openadapt_ml/experiments/demo_prompt/run_experiment.py,sha256=uchhadnqxauBXxlalTh7wXPLOOXk7NBY1mwXiN7rpHI,16309
33
- openadapt_ml/experiments/demo_prompt/results/experiment_20251231_002125.json,sha256=08oryOF126toTQDN9xciodavvfsaWNnXuBs0aULwpfI,5326
34
- openadapt_ml/experiments/demo_prompt/results/experiment_n30_20251231_165958.json,sha256=u03VgYTQia_HzilzNjxdGLpUSdbo4SzmHqI-GXlvurg,26915
35
- openadapt_ml/experiments/demo_prompt/results/multistep_20251231_025051.json,sha256=FA1JgXXLor6on3lHlfJdNSuKzBca30ggH8IWSJEmmfA,11517
36
- openadapt_ml/experiments/waa_demo/__init__.py,sha256=9M8iLxO9GWAw-FIB-0tzsqaweLcO5EVP1Sc5BoK16iU,363
37
- openadapt_ml/experiments/waa_demo/demos.py,sha256=UwO0EYy8wUEggaBaI_cXuYe_jwSB1hx3ZtPf-z9bhjc,13796
38
- openadapt_ml/experiments/waa_demo/runner.py,sha256=OxgQhZIqhYeGDYmAcQLEsFh3B053rnuPL0ZEIoXz0bI,24327
39
- openadapt_ml/experiments/waa_demo/tasks.py,sha256=jw1QwbOt8xmWBW2lmBWcJzKBXssjv_e0j49MlC2rVJY,5425
40
- openadapt_ml/export/__init__.py,sha256=mKehKHOio4jGcK-3r0-pb446GdKMPs0O9hAu4S0_R7s,266
41
- openadapt_ml/export/__main__.py,sha256=0ObtWcdzf6p7gPwhNlCKpNm2FIhmusdYNkuk8tyt77U,149
42
- openadapt_ml/export/cli.py,sha256=goTKNq9cOO9wsdNluLMH_-f9kdWShH3FPP8sCZ6KaPI,2331
43
- openadapt_ml/export/parquet.py,sha256=1BGHJKJc302trn64mUbuhxYMBPNvH80QW5f8GM63njk,9664
44
- openadapt_ml/grounding/__init__.py,sha256=uMvcALFRXmKD6PHhqLZ24Y6zhRUs46_PnWYqiqJP5cM,1412
45
- openadapt_ml/grounding/base.py,sha256=mnjT25nxltZCD0VBzgIgj2kuCcB4sgXBN97MBaW5P6c,7688
46
- openadapt_ml/grounding/detector.py,sha256=z-6Y_jnUNnhviUjKv6okjJ0r13DmBiNZMzNJo0rTlBY,19786
47
- openadapt_ml/ingest/__init__.py,sha256=7YASU-pOXtjalcRJ3WCbGuEWne0aVZtypsfcEmp-yFE,1437
48
- openadapt_ml/ingest/capture.py,sha256=SR22U6M7hU_QAsCYAG4JaxGFHpLZg96k_SJ-tkXAgbs,10218
49
- openadapt_ml/ingest/loader.py,sha256=PPBFMA9d7oc2bjXgrdPCZQBnv3MvoDFQNJvXeI90_j8,9865
50
- openadapt_ml/ingest/synthetic.py,sha256=DSUyqbLxHtJjELitTP2C_3bv3-E0UW7P-RFAO9W8XFs,39302
51
- openadapt_ml/models/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
52
- openadapt_ml/models/api_adapter.py,sha256=9EfQrXuFYIT-ea-wmGzJlM4thgVVDIRZnqqS_gL-PbU,6541
53
- openadapt_ml/models/base_adapter.py,sha256=AG40BhdDORwUNYYg3DM1wsErX57aEJrkK0tyn0tEKhI,2050
54
- openadapt_ml/models/dummy_adapter.py,sha256=h4Zu-rjWgtG1r8jRtcsrX-FZm8iImrhrTQ7TsLfjE8A,1581
55
- openadapt_ml/models/qwen_vl.py,sha256=sIJUtDRXAcz9zh3uRWOMYVOxdWIXlcM3vazdNOAsY_U,17239
56
- openadapt_ml/retrieval/README.md,sha256=j4gXhTo6yH-5cuw4ER4174V-U6TQakOVT6Hj4kj7B0I,5696
57
- openadapt_ml/retrieval/USAGE.md,sha256=XDIrX-94Z5nC-wvnBY5yF5gTqUYixxCC3wwUFvQx5YM,9278
58
- openadapt_ml/retrieval/__init__.py,sha256=xocb84riKLUCezUioKssFRhAQsnvexh4W932o368_qg,2726
59
- openadapt_ml/retrieval/demo_retriever.py,sha256=fYPLKzlG7yGHfV-F-TECBbxkMyP1fqdqPzTJ7G0oXYU,29184
60
- openadapt_ml/retrieval/embeddings.py,sha256=W4Bqo48Ds4BI9zZg1awkSi9p5kplRRALEjEgVh-jbsY,19239
61
- openadapt_ml/retrieval/index.py,sha256=UBFnSxp5T5eKt2txFcd0FytKCw1qxONZfxnFJVrduRQ,5710
62
- openadapt_ml/retrieval/retriever.py,sha256=idJcz4pUHgPHuZvX3VIUmO8Vs-iw4_1w6UUypynRfVY,4579
63
- openadapt_ml/runtime/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
64
- openadapt_ml/runtime/policy.py,sha256=M-OyhaE5gWh7e3KJ0Ip_YavlfRaEO7S4UKNjvqH_qsg,6724
65
- openadapt_ml/schema/__init__.py,sha256=W1Rx58WjFpUE2D1hdujD6tkxr5m8U3nE4JBMwdw7kLc,2787
66
- openadapt_ml/schema/converters.py,sha256=3qK1z8EATFaDi1M0w9T7PLiRtJu6OtQM7JG7qbE4EKU,18392
67
- openadapt_ml/schema/episode.py,sha256=_QQ34V39DYLaOx5GnH4mKHoXteekqRSq1C2aJe_Y_5Y,15399
68
- openadapt_ml/scripts/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
69
- openadapt_ml/scripts/compare.py,sha256=IUi3lQfOV8qm1JMvPDAnk3UqT7vPYNjLGeN8qda1wXc,56921
70
- openadapt_ml/scripts/demo_policy.py,sha256=NqB1akCKT5dlLER5ToBpF15QYaWnbXtRgWBvrOb6HJc,2105
71
- openadapt_ml/scripts/eval_policy.py,sha256=Tk1QMNlkfcyHToAqdTgdLkJyEGyVLXMcE5EVmJb89Ng,10381
72
- openadapt_ml/scripts/make_gif.py,sha256=H9fevBZFH31_7vs-OROfg9A2U6eboWXUNjyU2-XLMqw,4439
73
- openadapt_ml/scripts/prepare_synthetic.py,sha256=RbnY3QiH_RFk2du3awXFn90jxJQNUkhWlei2NSQ1aUs,1109
74
- openadapt_ml/scripts/run_qwen_login_benchmark.py,sha256=NWIhCAFSX5pYKFRCec7RkrYtzvz2LNMqhDfXcKxlagM,5655
75
- openadapt_ml/scripts/train.py,sha256=BrDJxiZx1S8igpNi6hC287kh51qU5RtPcdVq1yxEJkQ,6685
76
- openadapt_ml/training/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
77
- openadapt_ml/training/benchmark_viewer.py,sha256=iD56K467n0qS-nrcxxGZtABQs6qswB2je5Uj4xEacNI,174238
78
- openadapt_ml/training/shared_ui.py,sha256=7ZdBDo-__pHUMiuvSot-waYvpinAullRXQTsiPoruBw,4823
79
- openadapt_ml/training/stub_provider.py,sha256=fw2b0EzsCDDFhTZQIKR49BEFmSEN1d4g9rhFihzErWY,10508
80
- openadapt_ml/training/trainer.py,sha256=K4LMDe-GxnGZB9gt0IGMU67DcqtcggaaQIbJp4GYcYc,90713
81
- openadapt_ml/training/trl_trainer.py,sha256=9NQnNt2MDLUUUA_V8f2PwsDLeHnb2fOnmxol24-PyjU,11355
82
- openadapt_ml/training/viewer.py,sha256=7uA2SHW1Uh7v65s8sOSaYuAVUlyp6GsCOGy6YA0hnBQ,121132
83
- openadapt_ml-0.2.0.dist-info/METADATA,sha256=-pGXAMvRsWMHCdjsj3GzALIcAr0fPaMvEaUv8jkGYfA,28839
84
- openadapt_ml-0.2.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
85
- openadapt_ml-0.2.0.dist-info/licenses/LICENSE,sha256=2E5UY67RVLedJuNnwGudkAMtfM3LZNUcHgmaL89TAfw,1068
86
- openadapt_ml-0.2.0.dist-info/RECORD,,