cogames 0.3.49__py3-none-any.whl → 0.3.64__py3-none-any.whl

cogames-0.3.64.dist-info/RECORD

@@ -0,0 +1,159 @@
+cogames/__init__.py,sha256=Loao4kW7AqCRVKcQuA7lKEsyKl9k_Zf1oVlPXvgqjbw,23
+cogames/auth.py,sha256=W7BgRhrCTKob68p6HHTKVFjPayySaBG03eIgcGjw_jI,15852
+cogames/curricula.py,sha256=a7Nd-av4epAjmoqlefyU4q5eAitpYjCEGDCF_tS_iaY,825
+cogames/device.py,sha256=GVC7g4tNVySn_rSbHJB0jGKvpYBL8-VmeYEQXWXtvy0,1680
+cogames/evaluate.py,sha256=xm4dCLAXZFhfygDeC_617QXRa5DQYB_6EENEeeDAiWQ,9287
+cogames/game.py,sha256=Qpyd4UOv97S-PCcEb422XFVbuXk46qhEx7_FlAr8ayA,3331
+cogames/main.py,sha256=zYiY_cB0kiRproN4-Xa6MSwlff3-9hpFZLOZ-YfSwEg,79367
+cogames/pickup.py,sha256=gCm-BLMSKN4iflyn2EdsxE7ywKYzMgjozep5o_306yE,6545
+cogames/play.py,sha256=SV_gKg5kNj68m7bavmEYomLmmWxVGbc50ZhwAEMDy5E,8014
+cogames/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cogames/train.py,sha256=C4ZP00fxHspiQK-p00EPXyfcwiFKNIBUF1iBe2EE0GE,17894
+cogames/verbose.py,sha256=WcbVMpB43YpNJxZcwEkFtIqiIIxPLMZMu_3XQmeY59w,2916
+cogames/cli/base.py,sha256=UpMqeJ0C8i0pkue1AHnnyyX0bFJ9zZeJ7HBR6yhuA8A,54
+cogames/cli/client.py,sha256=UFWn6Kjbp6nCYjeC5wMwIJAM5o_hngNdZ_yv24aeiOI,9025
+cogames/cli/leaderboard.py,sha256=eIbiSBOkvoi5IL2weiKBOJAjWGHquDk1PWoHKA8uu3g,13941
+cogames/cli/login.py,sha256=_i1Hdbp_wAMsX0NLbCTSr7GmOFbzSqwyNP79OqLOg40,1237
+cogames/cli/mission.py,sha256=Shd7wivVdRbWTZ02yH334QZfwAqNy6mvRzLvU01iJrA,21674
+cogames/cli/policy.py,sha256=JI_aX0lcY-kaZWl-RhUfFu0vknF2mXHS8y0CQRRhpao,8743
+cogames/cli/submit.py,sha256=vB8EhB1z-SOUIdRhIkMEm_sbjFXbarVUn5BuiPa7SMI,20927
+cogames/cli/utils.py,sha256=K4eb-RL73HtavNGxZjn_SVDLAwfFamLCEQNTCNVdmPU,895
+cogames/cli/docsync/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cogames/cli/docsync/_nb_md_directive_processing.py,sha256=A920oS21mw9ZU2vp-cILzq5Sa-AMtmwMSayZKcQwX9M,6365
+cogames/cli/docsync/_nb_md_sync.py,sha256=oa7P-7OOPUtkKFX6C7SD4fk8ufGDNFvi__zKfhkw4rw,3942
+cogames/cli/docsync/_nb_py_sync.py,sha256=BfnfJ-1cRjmpzvggiR8E4vQkTak7AHm36pIK6xJVy-s,4578
+cogames/cli/docsync/_three_way_sync.py,sha256=fug-HXNYqJhJvusWHnGmLQewtiAUKhCBOd1Tuu7__sA,4877
+cogames/cli/docsync/_utils.py,sha256=zPfXzVa0XbqZReGvMUvxl4yLWc2oFeJJiefOBjOHaCA,2616
+cogames/cli/docsync/docsync.py,sha256=dtGG91Co89VdiwRwKk57kYd-qnHiFIgIkS1OI12AYyQ,5664
+cogames/cogs_vs_clips/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+cogames/cogs_vs_clips/cog.py,sha256=P6T9y6iZ4aAXATqmQ3ooRLvP5K60r3o-viX4eoiaZq4,3201
+cogames/cogs_vs_clips/cogs_vs_clips_mapgen.md,sha256=Tof8NFILYAj4nTYXFXgbGlLaryHj0Zvjdm-lBGKP5dE,13378
+cogames/cogs_vs_clips/cogsguard_reward_variants.py,sha256=V54Ly7T84cRfBoHv7Am7tv2OpL43Ksm95Fqkwy_TWAk,5819
+cogames/cogs_vs_clips/cogsguard_tutorial.py,sha256=_J4yCpf4783Cp4Z0dV6aRY0DCWDxQ4P4PBVhePRMQIw,1623
+cogames/cogs_vs_clips/mission.py,sha256=KTpfBPiwzAxobmnQ-WqPn0dLiAR4goFeHUAeyO4ahgA,12480
+cogames/cogs_vs_clips/mission_utils.py,sha256=-tB8JclTaq8Zfh8V5DPor1A3Ncuq8ndsTjgVnIxUV7s,671
+cogames/cogs_vs_clips/missions.py,sha256=ZAmS7qYViKwF4XazyqE7BE5HEhgBhwWhWCCYBfUCU7Y,2430
+cogames/cogs_vs_clips/procedural.py,sha256=M_a6uRsHzatVSW8PrVTAupv2u1M_ehOgM6XnEB3_Gbk,25947
+cogames/cogs_vs_clips/sites.py,sha256=Mq10ammCu49-xngQtJy8UwfRR6d78FN2dzxxi7HZy04,4916
+cogames/cogs_vs_clips/stations.py,sha256=HxTPQnY2Y8kqK42OnaSAg8S69yO7g2pr-WkGiaL967I,9174
+cogames/cogs_vs_clips/tutorial_missions.py,sha256=p2ds5t_cfD9xyFLSq8A9xw-zO4-OHi9LSGdocl4JDC4,712
+cogames/cogs_vs_clips/variants.py,sha256=zU32g8Hy2GEBgOMjf6VU6Phn4X7wVvfV4_shjtddOc0,11783
+cogames/cogs_vs_clips/evals/README.md,sha256=aglXp04_1AXwxk-zfYsqOi3qbVpSRvOVJY3j5I_yCSA,7048
+cogames/cogs_vs_clips/evals/cogsguard_evals.py,sha256=OgIFjQIUbEbjyWxalr4_eRPYoi3bkgvFNsZfW9kKu7I,2510
+cogames/cogs_vs_clips/evals/diagnostic_evals.py,sha256=RakNmoy8I5lgYqlUT38sctIszaLWLQmnekaNjg---Ck,13626
+cogames/cogs_vs_clips/evals/difficulty_variants.py,sha256=SCUAbbOCCYuK2gEcuJMWA-qAlQfme9AjEOIESFj6DBY,4319
+cogames/cogs_vs_clips/evals/integrated_evals.py,sha256=wTq2JOeF4pxU59TFn0VLs7atNiVG4VFfmF-z8sFuERY,2784
+cogames/cogs_vs_clips/evals/spanning_evals.py,sha256=CWa0HtG6LkxVP-SDDeWCSOC9Tw_20fFoAtQurZHJzTo,7516
+cogames/docs/SCRIPTED_AGENT.md,sha256=MDvYQ1Xccexnr_HjCnF5hNsZtqGAy2s0VA3tdgZHego,10918
+cogames/maps/canidate1_1000.map,sha256=DjLOgWDX9e9bmnA7oXcyIR-yilpoWufW0tsU12wSMmI,1023273
+cogames/maps/canidate1_1000_stations.map,sha256=RXqSceZiEf6TmEfeFO8vsNiAnyhox4zCLp77xPwSO7g,1023402
+cogames/maps/canidate1_500.map,sha256=FEhoftpEy9m4tgPwKKK6aZZp_gsgAbxZpMGWYMdtdls,261773
+cogames/maps/canidate1_500_stations.map,sha256=IdmnGb0djh9XzwCalJJ51sa5dXsyC5_T7r12w8t_nEo,261902
+cogames/maps/canidate2_1000.map,sha256=g6l7lnXEtg0TCEsLzY3WZ2opAa7nZSGUqnQdy8IpZGo,1023273
+cogames/maps/canidate2_1000_stations.map,sha256=t6pCH0EZ75VRPTOgtNbIfQ0AmSaxhjRpiXMbCCsnk0M,1023402
+cogames/maps/canidate2_500.map,sha256=zLOu4--t8BTDawScdt9eTDHnVugUBRp-6FBn0RtvCoU,261773
+cogames/maps/canidate2_500_stations.map,sha256=jyL7DKfmxrogybxa6abJa7ClM_xWq30hvRMmsyvDUK8,261889
+cogames/maps/canidate3_1000.map,sha256=AMNljeavU6z15Q4Ts4w9QtgCnPj3S0AcEncEaHKL_0o,1023273
+cogames/maps/canidate3_1000_stations.map,sha256=AENVFj_oQ2UBNiWeBWUKkM3QR-9M_vRO-TdxBh3KDoQ,1023402
+cogames/maps/canidate3_500.map,sha256=Tu_cTJWz1KRso48wtLHAa7-6M5PAltOCMGwnNDSK8co,261773
+cogames/maps/canidate3_500_stations.map,sha256=99QB-4o4jh8SKTkNryh7QuK7X71iIX7DeRYgQ8MMA38,261902
+cogames/maps/canidate4_500.map,sha256=Jn0CBxX5iwGHBlV7u-xyAQAuIdlUSzMYCSN1F_BvpLk,261773
+cogames/maps/canidate4_500_stations.map,sha256=GBXWYYjEVk2N2JDeRs_cowfXRTjEG2m18PQYILLJ-pQ,261902
+cogames/maps/cave_base_50.map,sha256=jHZyZfEJ9awJS2cwK5sWjTN23irhhyDeb4gWirClvzw,4052
+cogames/maps/machina_100_stations.map,sha256=oJqPvGVpLyqm35YFgJ2qLfdA2PW1ww-4a-m6JlDdshY,12702
+cogames/maps/machina_200_stations.map,sha256=svhatiFmKPq08u1_xeiCuBhkJKq3ucTSn8k06_yMWik,45002
+cogames/maps/machina_200_stations_small.map,sha256=7OVVBCOCWCIpJkaCG1uxDQUDn1aFrrImNX92wOMLukE,5927
+cogames/maps/machina_eval_exp01.map,sha256=ep5_8N8JyDMVRK2YrAPFbrqiR5WsVKkJPkhTxYxtOqE,2073
+cogames/maps/machina_eval_template_large.map,sha256=gZVIICxthSDnzEa8rT5EHR63XapT1-4xa6sQcsYl1mw,4823
+cogames/maps/machinatrainer4agents.map,sha256=O3dzwawhcXf72CPlDY4u0aN2M9sZ2W_vgte-hIKoA8g,5382
+cogames/maps/machinatrainer4agentsbase.map,sha256=7dUwTtCJK4Nz5rEgXVKzEx3xiZb8X3euXYZNj55nw0E,5382
+cogames/maps/machinatrainerbig.map,sha256=X5DkcUflf8mAt9X_rtatsixekwX2H1JClLGCG2Xp9iQ,51348
+cogames/maps/machinatrainersmall.map,sha256=F8ndClYV6lw5p0XMe8hFp_ip_zzPE2-ynCX7oeSwQKA,12476
+cogames/maps/training_facility_clipped.map,sha256=3xMzGq085cPBOmlSig8OTCOsWXKJ7c667k3wEj2gaHc,1358
+cogames/maps/training_facility_open_1.map,sha256=y48S99f5YiVNbvTfuSlfAImloH7IsAMi5LxwircClKI,776
+cogames/maps/training_facility_open_2.map,sha256=FMYqOC9ayQyOID6b6-ZgKHyKBuEyhhkfbEHo0MlYv5A,1326
+cogames/maps/training_facility_open_3.map,sha256=cc_whap2-U6zvxGGOXeUIYw4_tCR_0V4mMd1YmfMeJI,776
+cogames/maps/training_facility_tight_4.map,sha256=xH1RvLFt6XWdeAaWjyVQdSr-KkThAvQO47kaXhg5DkU,776
+cogames/maps/training_facility_tight_5.map,sha256=aVF66xgV0Df-NYFkXopabn30mPt_ZRQVER4v5ycm9F0,776
+cogames/maps/vanilla_large.map,sha256=hnlo1w8mAnPTl_LT1h-MW66_bplNiJ4OSFw13zG0yu4,2062
+cogames/maps/vanilla_small.map,sha256=P1l1tNL68Z8U3gFSwAIYsJzgx7kWz0O_UxfCfeeMFOs,590
+cogames/maps/zelda.map,sha256=1_rwezA27n5V9R3JY7WXniFVm8d31_Y1kyr9vn_gPjg,22387
+cogames/maps/diagnostic_evals/diagnostic_agile.map,sha256=tFRCId6FyEPCuaVEeSYySHcDH3rq29KLiAz2a1vANHM,812
+cogames/maps/diagnostic_evals/diagnostic_agile_hard.map,sha256=sW019536JKJBeVKT5T1mWaeqaOBttlRJ2IpEQjO5jc8,1540
+cogames/maps/diagnostic_evals/diagnostic_charge_up.map,sha256=g9Nvtgb7kCGcKr7b387NmVOLWbEeGJTyizRzx-MEfBc,831
+cogames/maps/diagnostic_evals/diagnostic_charge_up_hard.map,sha256=wj0vlpDQETSmYpcw0bmJDDqcQ5QbPNnoWBqBZibZKlc,1701
+cogames/maps/diagnostic_evals/diagnostic_chest_navigation1.map,sha256=T12a6zUgVfsJhQyvL0kNCx8vWU_FdJCoa75ZVXCi3xU,1763
+cogames/maps/diagnostic_evals/diagnostic_chest_navigation1_hard.map,sha256=1PzBOT1j5R0eT1glOXop01UdIZtMOTEHTaL28cjKmKU,3096
+cogames/maps/diagnostic_evals/diagnostic_chest_navigation2.map,sha256=17GVIQs5mJEywC0tLV1y3uyEwIpVomnkhEFeNebCAl4,1041
+cogames/maps/diagnostic_evals/diagnostic_chest_navigation2_hard.map,sha256=NmLbVM90xRpHJLiPPSeU8puFXUZ7eQougPnjsVse_3g,2770
+cogames/maps/diagnostic_evals/diagnostic_chest_navigation3.map,sha256=66zB8rAv85ZiPDq6dYto54g2XCaBIV6m1qqAFS2yWKM,1123
+cogames/maps/diagnostic_evals/diagnostic_chest_navigation3_hard.map,sha256=hGJUVN7igzbuPoD9tRu8GCAsn3ea6T2CAwcFPrjIW7w,2636
+cogames/maps/diagnostic_evals/diagnostic_chest_near.map,sha256=FN7nA1QpIkg454buDSuIhYTrKJ1zmS_0S_cs--jTf6c,1798
+cogames/maps/diagnostic_evals/diagnostic_chest_search.map,sha256=NgWr6ZZ6z5caOdZAxAIXuNwLKnbShKUcC4J7adm0xxc,1798
+cogames/maps/diagnostic_evals/diagnostic_chest_search_hard.map,sha256=rgn6kw3-pYWS586UK7ecuSWcU-tNCsC63gPHg51PAlE,14378
+cogames/maps/diagnostic_evals/diagnostic_extract_lab.map,sha256=w6k0Srm1Z3P7oAHzIab1uUlVNUTrxi99VDA1rjgYZMk,642
+cogames/maps/diagnostic_evals/diagnostic_extract_lab_hard.map,sha256=IWwi5emOlfR2h3aX4lzNVBlIlcWFlsktUMYiws71cdE,1512
+cogames/maps/diagnostic_evals/diagnostic_memory.map,sha256=TWMLViQT9b4pYlWgpZUlO-7Z1CGkNfFIg4Y7vboBng8,1210
+cogames/maps/diagnostic_evals/diagnostic_memory_hard.map,sha256=pY4KyQ56YYWFkAWVHmegoO9KdGL5WjsMFPYy6rhF2s4,1992
+cogames/maps/diagnostic_evals/diagnostic_radial.map,sha256=RxVoYmm6U_qUwmmgXZqB2ck3q6fk0al7P5CwU41wdas,3045
+cogames/maps/diagnostic_evals/diagnostic_radial_hard.map,sha256=PPAxa97Eb5LQaDlKaPhJoWxnw31ADLh_soXhOTJ-zlA,24525
+cogames/maps/diagnostic_evals/diagnostic_resource_lab.map,sha256=CEZlD1OGnOP9VEksGbxnfSWAZ-JV1hrGF16FtDJTRY0,1798
+cogames/maps/diagnostic_evals/diagnostic_unclip.map,sha256=Qj3Slpl7hCP5Xa84A1Ek6wWl0dI_8U7Etm1DMitDyxs,1798
+cogames/maps/evals/eval_balanced_spread.map,sha256=Ix-NFCndwvJds3JTSam7HtYJjiuXJRoR8c33SDwUdZc,2162
+cogames/maps/evals/eval_clip_oxygen.map,sha256=a3RqOglzbNScOhsN-WDBuZY0CgmgFwKgSwTginmpbT4,2164
+cogames/maps/evals/eval_collect_resources.map,sha256=6rQpYwqDHR-bmA3sW-XHBDP9n09uorv8yQLTaBUWmfA,2164
+cogames/maps/evals/eval_collect_resources_hard.map,sha256=6LLWYSvdByGqqi-x9-w-akbpz_I--upLTTfmy7uXRvU,2164
+cogames/maps/evals/eval_collect_resources_medium.map,sha256=avNWaX3TfSxyKveRb6dfymUKq390PXzGvP1D0c4GlN8,2164
+cogames/maps/evals/eval_divide_and_conquer.map,sha256=p_Aw_g2GoAD1zlf45FR4fNjvBAdngpaqV6KZmEiFxxk,2162
+cogames/maps/evals/eval_energy_starved.map,sha256=1R5K0XHFHG_sJF3jgDQt5vkEd2S0Xoqc5H3iSq_4QVQ,2162
+cogames/maps/evals/eval_multi_coordinated_collect_hard.map,sha256=2ED8sqV9RHPdFEo5F3DBUwC9T0DqcGNTmJbF9iprMLY,2165
+cogames/maps/evals/eval_oxygen_bottleneck.map,sha256=XlUTquPT4V3YpbVc5sLpQBm8HVTeeq0W6pRX-dTpXEU,2164
+cogames/maps/evals/eval_single_use_world.map,sha256=Hj1KX5Fa4FAmhI8wIR2opRXqn5uUFbDLRthkiZA0ie4,2164
+cogames/maps/evals/extractor_hub_100x100.map,sha256=knCtQaQjHV5hwOn95JEfXbusB9NvODGG2u6lIHKZmSY,12792
+cogames/maps/evals/extractor_hub_30x30.map,sha256=RM4ZDkThW1emXOPwyKBr_3TB1JsQ0_NVSr7Ma8AHLdQ,2082
+cogames/maps/evals/extractor_hub_50x50.map,sha256=Y3BhaSqUNXmas-5ESWVmceeKIX4tbHdbCHG5UkWLvy0,4142
+cogames/maps/evals/extractor_hub_70x70.map,sha256=nB-GV-DDDZcUZLciCga5WJJYh932M8xgL7aZiU1OwmI,7002
+cogames/maps/evals/extractor_hub_80x80.map,sha256=hPL2bjISsayK7slBPQkGyQBvkDkWFrxd7jbNSNupXE8,8732
+cogames/maps/planky_evals/aligner_avoid_aoe.map,sha256=9C2pJsASv7rVUfqpLdoMTA1K01wGRJrBXBvbRRXaGqE,648
+cogames/maps/planky_evals/aligner_full_cycle.map,sha256=0ONRKlUI7XSI3L6PhglzNHup_SuNHzsHxl9bNYdmxUw,604
+cogames/maps/planky_evals/aligner_gear.map,sha256=5bK9MwJ3HGzB1sWmRol5AdxZ9rwjPp6-OsfFPPQSdDw,460
+cogames/maps/planky_evals/aligner_hearts.map,sha256=oCF8DpmxyqpIARzlQRwAEZs3mMvy-KMPmVnVAYn---E,474
+cogames/maps/planky_evals/aligner_junction.map,sha256=NzkDOi9lGOpR2avkEPaK2jZcq3x1OBnM-AWc_hTzwUI,542
+cogames/maps/planky_evals/exploration_distant.map,sha256=CMk3M5uzrCUZjdow9_DErD2vJzdLe1e9idsFKkMcxZo,648
+cogames/maps/planky_evals/maze.map,sha256=cJDGu2b9JHOxWQgeXQsGWKO9xN_CGjbeOeypAehPfUY,662
+cogames/maps/planky_evals/miner_best_resource.map,sha256=I5MT8ChAlFu6fXWnD3yqQvqFax9KRjRnCYDTd0hyAAY,542
+cogames/maps/planky_evals/miner_deposit.map,sha256=8DgTMdxpvJ4QAbNyu90twhFKLjnxav0k7sRzTNTEf5c,474
+cogames/maps/planky_evals/miner_extract.map,sha256=XUgtZaXsnA6rcE2W7DBPjQxS4eUsYC3ZnX6IjzY6Ljk,488
+cogames/maps/planky_evals/miner_full_cycle.map,sha256=38nPZ8TujgWIUFzBlZgwDq_Ylr2jlglczgxV53joa7o,560
+cogames/maps/planky_evals/miner_gear.map,sha256=PVh6xU8gNKy2XwDv9Vv4ZaX_-0UtGO3cDQg4EQLVWH4,446
+cogames/maps/planky_evals/multi_role.map,sha256=8HTHlWxWvsXSbpSU9U0Polq08MOdgKLZMkwndXmsqmU,604
+cogames/maps/planky_evals/resource_chain.map,sha256=NWrOKQkcezvZ9LK4Fz26LOYTlLJD2e6AyKk_gxQdNP0,622
+cogames/maps/planky_evals/scout_explore.map,sha256=AiYUNrRBZpI6t-ObSx3yZLAIV1ofdNe9gXzn07U9AF8,662
+cogames/maps/planky_evals/scout_gear.map,sha256=DoC3Dulxih20js67PuE9mXEiLy3kFafZNuHVWaKzjB4,460
+cogames/maps/planky_evals/scrambler_full_cycle.map,sha256=fkmuibPalfsUUMXtawMLt3FquWmJEPb6JdLlTUuDn98,604
+cogames/maps/planky_evals/scrambler_gear.map,sha256=WwUCa_kKUQqPdsS_UG2KgPtZIlBd9sUejo0ls8u7lL4,460
+cogames/maps/planky_evals/scrambler_target.map,sha256=HwwSIvQvGuEmge3MrzLViPZRuWHNcwY6ZFp134GIN30,524
+cogames/maps/planky_evals/stuck_corridor.map,sha256=8wMEjZkZQXevtl2uYKUjFZnln5PWzuePqYRCAooyc_I,572
+cogames/maps/planky_evals/survive_retreat.map,sha256=bf2bWTZQ6ZY1RPMrRQC2lyMufCfGiJ-VvDI_A2aiS4s,542
+cogames/policy/__init__.py,sha256=JFjmvqYzj_j_mQ4Bmqmc60KU49HF63i0s8lIEu1G_b4,116
+cogames/policy/chaos_monkey.py,sha256=FBUK5kZj5Kzgeo8VJ61iFk59AHLLE7Y0bvE7ffPIiHE,1825
+cogames/policy/pufferlib_policy.py,sha256=5JaPTLSsicWNCNWYSiJnbijOCKpjkyie4KKvw0-awYE,5617
+cogames/policy/signal_handler.py,sha256=19_i72AlYZmMifBDvZuaqLVrFIbWLa8mW-7In7nbG3c,2060
+cogames/policy/starter_agent.py,sha256=7jRPOiNLHsR6Nrz6zy09YP02zH87PTQ9ogQadCczd68,7680
+cogames/policy/trainable_policy_template.py,sha256=GbB4vi0wsIiuh_APtfjnavvh9ZVUx4SBmBDhU6Msx7s,7337
+cogames/policy/nim_agents/__init__.py,sha256=q22rmDMKHMO2A4Cqt0P-kuH3f6yaldlaBaRuGydeEKM,1247
+cogames/policy/nim_agents/agents.py,sha256=2uxzs8yxcz3obdA_gaWf0qgojFbXSIU0W3cS-9XD0AI,5038
+cogames/policy/nim_agents/thinky_eval.py,sha256=NqIZng30MpyZD5hF03uoje4potd873hWyHOb3AZdeZk,1154
+cogames-0.3.64.dist-info/licenses/LICENSE,sha256=NG4hf0NHdGZhkabVCK1MKX8RAJmWaEm4eaGcUsONZ_E,1065
+metta_alo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+metta_alo/job_specs.py,sha256=_QT23tGb7V8W1jqiUijeQUQh6ZvKfW5-KYiYSuNTdVI,450
+metta_alo/policy.py,sha256=RnBwDzQ7abFiQSJe6MYr_JtsyHlKBES1CkDN6mZuvQI,613
+metta_alo/pure_single_episode_runner.py,sha256=VsSBEn2BVWWHtiZ0SdL9_QMTom_vbbIyPohRVa9rjTc,2720
+metta_alo/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+metta_alo/rollout.py,sha256=hc7JRQD4ilfqTO57BvKav2v_JW1Vbcc-oYcEmoWVlB4,10192
+metta_alo/scoring.py,sha256=7ikuP6Vnvv57bAc_ARbTadDSf6wm5dCY49xX9QjD_fs,5809
+cogames-0.3.64.dist-info/METADATA,sha256=Vnhu2_9kNcq7ALn4VF4VFMP8j25Mj6ZE1f-d1p7zBU4,234991
+cogames-0.3.64.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+cogames-0.3.64.dist-info/entry_points.txt,sha256=GTHdgj-RC2hQYmjUMSp9RHX8kbru19k0LS2lAj8DnLE,45
+cogames-0.3.64.dist-info/top_level.txt,sha256=YErBkYWJd3-eksLpbgbMrETni1MPBNL4mqEyhZUa0UE,18
+cogames-0.3.64.dist-info/RECORD,,

cogames-0.3.64.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2024 Metta AI
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

cogames-0.3.64.dist-info/top_level.txt

@@ -0,0 +1,2 @@
+cogames
+metta_alo

metta_alo/job_specs.py

@@ -0,0 +1,17 @@
+from typing import Optional
+
+from pydantic import BaseModel, Field
+
+from mettagrid import MettaGridConfig
+
+
+class SingleEpisodeJob(BaseModel):
+    policy_uris: list[str]
+    assignments: list[int]
+    env: MettaGridConfig
+    results_uri: Optional[str] = None
+    replay_uri: Optional[str] = None
+    debug_uri: Optional[str] = None
+    seed: int = 0
+    max_action_time_ms: int = 10000
+    episode_tags: dict[str, str] = Field(default_factory=dict)

metta_alo/policy.py

@@ -0,0 +1,16 @@
+from typing import Optional
+
+
+def parse_policy_identifier(identifier: str) -> tuple[str, Optional[int]]:
+    """Parse a policy identifier into (name, version)."""
+    if identifier.endswith(":latest"):
+        return identifier[:-7], None
+    if ":" not in identifier:
+        return identifier, None
+    name, version_str = identifier.rsplit(":", 1)
+    if not name:
+        raise ValueError(f"Invalid policy identifier: {identifier}")
+    version_str = version_str.lstrip("v")
+    if not version_str.isdigit():
+        raise ValueError(f"Invalid version format: {identifier}")
+    return name, int(version_str)

metta_alo/pure_single_episode_runner.py

@@ -0,0 +1,75 @@
+import json
+import sys
+from typing import Sequence
+
+from pydantic import BaseModel, Field, model_validator
+
+from mettagrid import MettaGridConfig
+from mettagrid.types import EpisodeStats
+from mettagrid.util.uri_resolvers.schemes import parse_uri
+
+
+def _validate_output_uri(uri: str) -> None:
+    parsed = parse_uri(uri, allow_none=False)
+    if parsed.scheme != "file" or not parsed.local_path.parent.exists():
+        raise ValueError(f"URI {uri} must be a file:// URI with an existing parent directory")
+
+
+def _validate_assignments(assignments: Sequence[int], num_agents: int, num_policies: int) -> None:
+    if len(assignments) != num_agents or not all(0 <= assignment < num_policies for assignment in assignments):
+        raise ValueError("Assignments must match agent count and be within policy range")
+
+
+class PureSingleEpisodeJob(BaseModel):
+    policy_uris: list[str]
+
+    # It is important that this is explicit, else the results will have to include the choices we made
+    # when randomizing
+    assignments: list[int]
+
+    env: MettaGridConfig
+
+    # For now, this only supports file:// scheme. Will eventually support https:// to send to s3
+    results_uri: str | None  # Contains EpisodeRolloutResult
+    replay_uri: str | None  # Where to place replay file. If missing, do not generate a replay
+    debug_dir: str | None = None  # Directory for observability outputs (trace.json, etc.)
+
+    # There's no way to ask us to generate a seed; the caller has to pick one
+    seed: int = 0
+
+    max_action_time_ms: int = 10000
+    episode_tags: dict[str, str] = Field(default_factory=dict)
+
+    @model_validator(mode="after")
+    def validate_output_uris(self) -> "PureSingleEpisodeJob":
+        for uri in (self.replay_uri, self.results_uri):
+            if uri is None:
+                continue
+            _validate_output_uri(uri)
+
+        if self.replay_uri is not None and not self.replay_uri.endswith((".json.z", ".json.gz")):
+            raise ValueError("Replay URI must end with .json.z or .json.gz")
+
+        _validate_assignments(self.assignments, self.env.game.num_agents, len(self.policy_uris))
+
+        return self
+
+
+class PureSingleEpisodeResult(BaseModel):
+    rewards: list[float]
+    action_timeouts: list[int]
+    stats: EpisodeStats
+    steps: int
+
+
+if __name__ == "__main__":
+    from metta_alo.rollout import run_sandboxed_episode
+
+    with open(sys.argv[1]) as f:
+        args = json.load(f)
+    job = PureSingleEpisodeJob.model_validate(args["job"])
+    allow_network = args.get("allow_network", False)
+    if allow_network:
+        raise ValueError("allow_network is not supported by the sandboxed runner")
+    device = args.get("device")
+    run_sandboxed_episode(job, device=device)

metta_alo/rollout.py

@@ -0,0 +1,322 @@
+import json
+import socket
+import sys
+import uuid
+from contextlib import contextmanager, nullcontext
+from pathlib import Path
+from typing import Callable, Optional, Protocol, Sequence
+
+import numpy as np
+
+from metta_alo.pure_single_episode_runner import (
+    PureSingleEpisodeJob,
+    PureSingleEpisodeResult,
+    _validate_assignments,
+    _validate_output_uri,
+)
+from mettagrid import MettaGridConfig
+from mettagrid.policy.loader import AgentPolicy, PolicyEnvInterface, initialize_or_load_policy
+from mettagrid.policy.policy import PolicySpec
+from mettagrid.renderer.renderer import RenderMode
+from mettagrid.simulator.multi_episode.rollout import EpisodeRolloutResult, MultiEpisodeRolloutResult
+from mettagrid.simulator.replay_log_writer import EpisodeReplay, InMemoryReplayWriter
+from mettagrid.simulator.rollout import Rollout
+from mettagrid.util.file import write_data
+from mettagrid.util.tracer import Tracer
+from mettagrid.util.uri_resolvers.schemes import parse_uri, policy_spec_from_uri
+
+
+class ReplayLike(Protocol):
+    def set_compression(self, compression: str) -> None: ...
+
+    def write_replay(self, path: str) -> None: ...
+
+
+def write_replay(replay: ReplayLike, path: str) -> None:
+    if path.endswith(".gz"):
+        replay.set_compression("gzip")
+    elif path.endswith(".z"):
+        replay.set_compression("zlib")
+    replay.write_replay(path)
+
+
+@contextmanager
+def _no_python_sockets():
+    _real_socket = socket.socket
+    _real_getaddrinfo = socket.getaddrinfo
+
+    def _blocked(*args: object, **kwargs: object):
+        raise RuntimeError("Network access disabled")
+
+    socket.socket = _blocked
+    socket.getaddrinfo = _blocked
+
+    try:
+        yield
+    finally:
+        socket.socket = _real_socket
+        socket.getaddrinfo = _real_getaddrinfo
+
+
+def _policy_specs_from_uris(
+    policy_uris: Sequence[str],
+    *,
+    device: Optional[str],
+    allow_network: bool,
+) -> list[PolicySpec]:
+    policy_specs: list[PolicySpec] = []
+    for uri in policy_uris:
+        parsed = parse_uri(uri, allow_none=False)
+        if parsed.scheme != "file" and not allow_network:
+            raise ValueError("Sandboxed runner requires file:// policy URIs")
+        policy_specs.append(
+            policy_spec_from_uri(
+                uri,
+                device=device or "cpu",
+                remove_downloaded_copy_on_exit=True,
+            )
+        )
+    return policy_specs
+
+
+def _write_outputs(
+    results: PureSingleEpisodeResult,
+    replay: Optional[EpisodeReplay],
+    *,
+    results_uri: Optional[str],
+    replay_uri: Optional[str],
+) -> None:
+    if replay_uri is not None:
+        if replay is None:
+            raise ValueError("No replay was generated")
+        write_replay(replay, replay_uri)
+    if results_uri is not None:
+        write_data(results_uri, results.model_dump_json(), content_type="application/json")
+
+
+def _run_pure_episode(
+    policy_specs: Sequence[PolicySpec],
+    assignments: Sequence[int],
+    env: MettaGridConfig,
+    *,
+    seed: int,
+    max_action_time_ms: int,
+    device: Optional[str],
+    render_mode: RenderMode,
+    autostart: bool,
+    capture_replay: bool,
+    trace_path: Optional[Path] = None,
+) -> tuple[PureSingleEpisodeResult, Optional[EpisodeReplay]]:
+    env_interface = PolicyEnvInterface.from_mg_cfg(env)
+    agent_policies: list[AgentPolicy] = [
+        initialize_or_load_policy(
+            env_interface,
+            policy_specs[assignment],
+            device_override=device,
+        ).agent_policy(agent_id)
+        for agent_id, assignment in enumerate(assignments)
+    ]
+    replay_writer: Optional[InMemoryReplayWriter] = None
+    if capture_replay:
+        replay_writer = InMemoryReplayWriter()
+
+    tracer: Optional[Tracer] = None
+    if trace_path:
+        tracer = Tracer(trace_path)
+
+    rollout = Rollout(
+        env,
+        agent_policies,
+        max_action_time_ms=max_action_time_ms,
+        render_mode=render_mode,
+        autostart=autostart,
+        seed=seed,
+        event_handlers=[replay_writer] if replay_writer is not None else None,
+        tracer=tracer,
+    )
+    rollout.run_until_done()
+
+    results = PureSingleEpisodeResult(
+        rewards=list(rollout._sim.episode_rewards),
+        action_timeouts=list(rollout.timeout_counts),
+        stats=rollout._sim.episode_stats,
+        steps=rollout._sim.current_step,
+    )
+    replay: Optional[EpisodeReplay] = None
+    if replay_writer is not None:
+        replays = replay_writer.get_completed_replays()
+        if len(replays) != 1:
+            raise ValueError(f"Expected 1 replay, got {len(replays)}")
+        replay = replays[0]
+
+    return results, replay
+
+
+def run_single_episode(
+    *,
+    policy_specs: Sequence[PolicySpec],
+    assignments: Sequence[int],
+    env: MettaGridConfig,
+    results_uri: Optional[str] = None,
+    replay_uri: Optional[str] = None,
+    debug_dir: Optional[str] = None,
+    seed: int = 0,
+    max_action_time_ms: int = 10000,
+    device: Optional[str] = None,
+    render_mode: Optional[RenderMode] = None,
+    autostart: bool = False,
+    allow_network: bool = True,
+) -> tuple[PureSingleEpisodeResult, Optional[EpisodeReplay]]:
+    _validate_assignments(assignments, env.game.num_agents, len(policy_specs))
+
+    for uri in (replay_uri, results_uri):
+        if uri is None:
+            continue
+        _validate_output_uri(uri)
+
+    if replay_uri is not None and not replay_uri.endswith((".json.z", ".json.gz")):
+        raise ValueError("Replay URI must end with .json.z or .json.gz")
+
+    trace_path: Optional[Path] = None
+    if debug_dir is not None:
+        debug_path = Path(debug_dir)
+        debug_path.mkdir(parents=True, exist_ok=True)
+        trace_path = debug_path / "trace.json"
+
+    with (_no_python_sockets if not allow_network else nullcontext)():
+        results, replay = _run_pure_episode(
+            policy_specs,
+            assignments,
+            env,
+            seed=seed,
+            max_action_time_ms=max_action_time_ms,
+            device=device,
+            render_mode=render_mode or "none",
+            autostart=autostart,
+            capture_replay=replay_uri is not None,
+            trace_path=trace_path,
+        )
+
+    _write_outputs(results, replay, results_uri=results_uri, replay_uri=replay_uri)
+    return results, replay
+
+
+def run_sandboxed_episode(
+    job: PureSingleEpisodeJob,
+    *,
+    device: Optional[str] = None,
+    render_mode: Optional[RenderMode] = None,
+) -> tuple[PureSingleEpisodeResult, Optional[EpisodeReplay]]:
+    policy_specs = _policy_specs_from_uris(job.policy_uris, device=device, allow_network=False)
+    return run_single_episode(
+        policy_specs=policy_specs,
+        assignments=job.assignments,
+        env=job.env,
+        results_uri=job.results_uri,
+        replay_uri=job.replay_uri,
+        debug_dir=job.debug_dir,
+        seed=job.seed,
+        max_action_time_ms=job.max_action_time_ms,
+        device=device,
+        render_mode=render_mode,
+        allow_network=False,
+    )
+
+
+def run_single_episode_rollout(
+    *,
+    policy_specs: Sequence[PolicySpec],
+    assignments: np.ndarray,
+    env_cfg: MettaGridConfig,
+    seed: int,
+    max_action_time_ms: int,
+    replay_path: Optional[str] = None,
+    device: Optional[str] = None,
+) -> EpisodeRolloutResult:
+    results, _replay = run_single_episode(
+        policy_specs=policy_specs,
+        assignments=assignments.tolist(),
+        env=env_cfg,
+        results_uri=None,
+        replay_uri=replay_path,
+        seed=seed,
+        max_action_time_ms=max_action_time_ms,
+        device=device,
+    )
+
+    return EpisodeRolloutResult(
+        assignments=assignments.copy(),
+        rewards=np.array(results.rewards, dtype=float),
+        action_timeouts=np.array(results.action_timeouts, dtype=float),
+        stats=results.stats,
+        replay_path=replay_path,
+        steps=results.steps,
+        max_steps=env_cfg.game.max_steps,
+    )
+
+
+def run_multi_episode_rollout(
+    *,
+    policy_specs: Sequence[PolicySpec],
+    assignments: np.ndarray,
+    env_cfg: MettaGridConfig,
+    episodes: int,
+    seed: int,
+    max_action_time_ms: int,
+    replay_dir: Optional[str | Path] = None,
+    create_replay_dir: bool = False,
+    rng: Optional[np.random.Generator] = None,
+    device: Optional[str] = None,
+    on_progress: Optional[Callable[[int, EpisodeRolloutResult], None]] = None,
+) -> tuple[MultiEpisodeRolloutResult, list[str]]:
+    if replay_dir is not None and create_replay_dir:
+        Path(replay_dir).mkdir(parents=True, exist_ok=True)
+
+    assignments = np.array(assignments, dtype=int, copy=True)
+    rng = rng or np.random.default_rng(seed)
+    episode_results: list[EpisodeRolloutResult] = []
+    replay_paths: list[str] = []
+
+    for episode_idx in range(episodes):
+        rng.shuffle(assignments)
+        replay_path = None
+        if replay_dir is not None:
+            replay_path = str(Path(replay_dir) / f"{uuid.uuid4()}.json.z")
+
+        result = run_single_episode_rollout(
+            policy_specs=policy_specs,
+            assignments=assignments,
+            env_cfg=env_cfg,
+            seed=seed + episode_idx,
+            max_action_time_ms=max_action_time_ms,
+            replay_path=replay_path,
+            device=device,
+        )
+        episode_results.append(result)
+        if on_progress:
+            on_progress(episode_idx, result)
+        if replay_path is not None:
+            replay_paths.append(replay_path)
+
+    return MultiEpisodeRolloutResult(episodes=episode_results), replay_paths
+
+
+if __name__ == "__main__":
+    with open(sys.argv[1]) as f:
+        args = json.load(f)
+    job = PureSingleEpisodeJob.model_validate(args["job"])
+    device = args.get("device")
+    allow_network = args.get("allow_network", False)
+    policy_specs = _policy_specs_from_uris(job.policy_uris, device=device, allow_network=allow_network)
+    run_single_episode(
+        policy_specs=policy_specs,
+        assignments=job.assignments,
+        env=job.env,
+        results_uri=job.results_uri,
+        replay_uri=job.replay_uri,
+        debug_dir=job.debug_dir,
+        seed=job.seed,
+        max_action_time_ms=job.max_action_time_ms,
+        device=device,
+        allow_network=allow_network,
+    )