mbrl

View on PyPIReverse Dependencies (0)

0.2.0 mbrl-0.2.0-py3-none-any.whl

Wheel Details

Project: mbrl
Version: 0.2.0
Filename: mbrl-0.2.0-py3-none-any.whl
Download: [link]
Size: 175545
MD5: 7685bab7d5aae3ccbc691e98b93a3474
SHA256: 1a858f5c43447668974fd0996ea1fcaa6fb0ad27e7494d4317b99151c317f29e
Uploaded: 2023-03-29 19:22:42 +0000

dist-info

METADATA

Metadata-Version: 2.1
Name: mbrl
Version: 0.2.0
Summary: A PyTorch library for model-based reinforcement learning research
Author: Facebook AI Research
Home-Page: https://github.com/facebookresearch/mbrl-lib
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3
Classifier: Programming Language :: Python :: 3.8
Classifier: Programming Language :: Python :: 3.9
Classifier: Programming Language :: Python :: 3.10
Classifier: Intended Audience :: Science/Research
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Requires-Python: >=3.8
Requires-Dist: omegaconf (<2.1,>=2.0)
Requires-Dist: hydra-core (==1.0.3)
Requires-Dist: termcolor (>=1.1.0)
Requires-Dist: tensorboard (>=2.4.0)
Requires-Dist: imageio (>=2.9.0)
Requires-Dist: numpy (>=1.19.1)
Requires-Dist: matplotlib (>=3.3.1)
Requires-Dist: gymnasium (==0.26.3)
Requires-Dist: jupyter (>=1.0.0)
Requires-Dist: pytest (>=6.0.1)
Requires-Dist: sk-video (>=1.1.10)
Requires-Dist: tqdm (>=4.48.2)
Requires-Dist: omegaconf (<2.1,>=2.0); extra == "dev"
Requires-Dist: hydra-core (==1.0.3); extra == "dev"
Requires-Dist: termcolor (>=1.1.0); extra == "dev"
Requires-Dist: tensorboard (>=2.4.0); extra == "dev"
Requires-Dist: imageio (>=2.9.0); extra == "dev"
Requires-Dist: numpy (>=1.19.1); extra == "dev"
Requires-Dist: matplotlib (>=3.3.1); extra == "dev"
Requires-Dist: gymnasium (==0.26.3); extra == "dev"
Requires-Dist: jupyter (>=1.0.0); extra == "dev"
Requires-Dist: pytest (>=6.0.1); extra == "dev"
Requires-Dist: sk-video (>=1.1.10); extra == "dev"
Requires-Dist: tqdm (>=4.48.2); extra == "dev"
Requires-Dist: sphinx (>=3.3.1); extra == "dev"
Requires-Dist: nbsphinx (>=0.8.0); extra == "dev"
Requires-Dist: sphinx-rtd-theme (>=0.5.0); extra == "dev"
Requires-Dist: flake8 (>=3.9.2); extra == "dev"
Requires-Dist: mypy (>=0.991); extra == "dev"
Requires-Dist: black (>=23.1.0); extra == "dev"
Requires-Dist: importlib-metadata (<5); extra == "dev"
Requires-Dist: types-pyyaml (>=0.1.6); extra == "dev"
Requires-Dist: types-termcolor (>=0.1.0); extra == "dev"
Requires-Dist: pre-commit (>=2.7.1); extra == "dev"
Provides-Extra: dev
Description-Content-Type: text/markdown
License-File: LICENSE
[Description omitted; length: 11833 characters]

WHEEL

Wheel-Version: 1.0
Generator: bdist_wheel (0.40.0)
Root-Is-Purelib: true
Tag: py3-none-any

RECORD

Path Digest Size
mbrl/__init__.py sha256=iOkfZoCoRgPcXVHJ5_AAMimpyOpZMl-mdcb0fPq_GIY 220
mbrl/constants.py sha256=D9Nm1_XGYmiOV9LwbTIldVC6iwOm_5M2ojgT8puzrhQ 318
mbrl/types.py sha256=vkYlwVY0iZ45Xf8sSINLiQbNxhgGqBJqb5MpIhfeSfI 2724
mbrl/algorithms/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
mbrl/algorithms/mbpo.py sha256=ifHBzJaHqB0bUgRubT_k4_Frgc3ttwCXjc-FIgCxfHQ 10740
mbrl/algorithms/pets.py sha256=2Bv4YwqFxoLwBFN_who3tXQTcACHaxHku99Ps7bLbuw 4561
mbrl/algorithms/planet.py sha256=ZWXyOYwfhKKrCapPmQEGSlIA99-AEm__k29m4rddKTY 6770
mbrl/diagnostics/__init__.py sha256=Rc6VnM6UEIsW5sR4RzGHBZEmzqwYIL_cky4jaVG2szA 350
mbrl/diagnostics/control_env.py sha256=ia1KApCflyda08IQczH-8AkvetSFnitc2l-kee-CbT4 7197
mbrl/diagnostics/eval_model_on_dataset.py sha256=opMCaTiEovCKMvFjriFsgtOKan0vyS0cjR8ChL-EDJw 4429
mbrl/diagnostics/finetune_model_with_controller.py sha256=J_6dKjFhBJyzFVBuJhd5NH94MYgBCRj0p_SMtc6Nh_A 4068
mbrl/diagnostics/planet_visualizer.py sha256=InGY9clziwcSdaXnVTmZFl_cLPG8CAsdUKa4NXKBKhM 5847
mbrl/diagnostics/training_browser.py sha256=8WYIRWZvMidOLxhzKvDjGa5TZlfxBMlv1Nfo-X3r8GE 12870
mbrl/diagnostics/visualize_model_preds.py sha256=QtUQxctWD1Cw_ZWbfZ3_WhyEJuJuEHTKVfGalj9HRpk 11127
mbrl/env/__init__.py sha256=3QKi7hHIZ30-jRC6qbJSxyfdX22Wchwf5jmPi8LGW_8 261
mbrl/env/ant_truncated_obs.py sha256=cAjVnH8nA3WcVu1M4ubYTqgLVIW4QyhMxP3FQZdsvj4 2422
mbrl/env/cartpole_continuous.py sha256=eGFyuq7gZWSZHcMRB3zAcsnMmywLcjO75Jm2Gb8DjRE 8176
mbrl/env/humanoid_truncated_obs.py sha256=-x2H_aDFun9zlltjralznKQTB_bvZNoxclugIvFxehI 2887
mbrl/env/mujoco_envs.py sha256=eqLXi4QRbV4cmWUOml92ooDGvl5IxkPo_GU4axqPFNQ 522
mbrl/env/mujoco_pixel_wrapper.py sha256=lFVIqdUu-eOexxpcLkKMdDmeSeiO_bXevEzjy115W14 4763
mbrl/env/pets_cartpole.py sha256=PQbH3lAaxHte5pjaPO3J5DIzxaIWDN0fcoCeloF_sM8 3273
mbrl/env/pets_halfcheetah.py sha256=K9igrhUfkGfdvjqI8P8eNOH8H8FEpKeR7PWbnljQVlU 4661
mbrl/env/pets_pusher.py sha256=Lx1qFlpi0CgRpEj_CKUBvKKc3ZjbntPZO0zxjrBOKVk 2439
mbrl/env/pets_reacher.py sha256=Zm7jthUIBJelf7V-SZTuBUNJ9pWlHWsl1bVHQm91vhQ 3869
mbrl/env/reward_fns.py sha256=X9LENzj5gAYEpVYF-IiaVXwjjyhXcQE1w8nOUYbZ8JE 1961
mbrl/env/termination_fns.py sha256=XEOpG0zNrGp59UJgtyOVC_cbO3Vx-oskPqMtXR4_W9M 2396
mbrl/examples/conf/main.yaml sha256=JakEbOzdXQurANXIpih2emfVLBIceXchmdf1T-rjDMc 479
mbrl/examples/conf/algorithm/mbpo.yaml sha256=judBe33Hp6WcVbUnGeDH1seLCAlpA29-s2watxpyqwk 1029
mbrl/examples/conf/algorithm/pets.yaml sha256=Hg6kHgZYTl0JMwGCFx6_jFQMI8E6PE7NkXSiiNgUz7E 492
mbrl/examples/conf/algorithm/planet.yaml sha256=VpGwr7yig7Y3Epta9Tldhi8-iqBZGKHJr5jP0XK3pP4 394
mbrl/examples/conf/dynamics_model/basic_ensemble.yaml sha256=ByASh9YzDeXSfNZFsGpYIydzCnci-FN7rsqlO9x7LA0 328
mbrl/examples/conf/dynamics_model/gaussian_mlp.yaml sha256=OJzo0pw6IsPNJ5v5yCCZa02lt8z5Zj-miJaeKINXpvs 209
mbrl/examples/conf/dynamics_model/gaussian_mlp_ensemble.yaml sha256=uHKf1_TKOUBCYxq23_uAXQLHb5UAtsJ0z46B4h49pxQ 297
mbrl/examples/conf/dynamics_model/planet.yaml sha256=UPJp3cQRB1necG-4nFD_1vwFXwVqwSjlbFvfytU4_ZU 474
mbrl/examples/conf/overrides/mbpo_ant.yaml sha256=JOSdKREjLzljR4Q2nfGCuSqPSkHOKGnv_QjiHxbbq64 697
mbrl/examples/conf/overrides/mbpo_cartpole.yaml sha256=yV8P5sxfu3aGdDCKM_O5rbqXBeZSR3Ym5r_izurTLKI 603
mbrl/examples/conf/overrides/mbpo_halfcheetah.yaml sha256=Lf-DXrq4c2VcihgMV6_8C7kQiwtoU0IiK5o3Q1mL6Jw 615
mbrl/examples/conf/overrides/mbpo_hopper.yaml sha256=ezybHQaEu2XLysZoXfxs_jLp_SM_FTZ7kJBOLynWrKI 643
mbrl/examples/conf/overrides/mbpo_humanoid.yaml sha256=j7g-vBXI1nLsU724laqasdx_5hMagRPzgUSSVtyhWJk 702
mbrl/examples/conf/overrides/mbpo_inv_pendulum.yaml sha256=TnqWmL3oBu8Ecn8dMjeark2DxsPycT4AHKNF3UZKOdA 618
mbrl/examples/conf/overrides/mbpo_pusher.yaml sha256=GPfzgtGRBa31w7v87r9YU3uDh0YPNc_4Psnm4kbUakU 623
mbrl/examples/conf/overrides/mbpo_walker.yaml sha256=1nUPgZ31VG7gLqBu9LgzkUDdUKb1GhrPIKWiLyE9imc 650
mbrl/examples/conf/overrides/pb_mbpo_inv_pendulum.yaml sha256=EwFnL5CBMROGJ0tkAtA2MaBwhXWA4o1arkY0P_vwtDU 640
mbrl/examples/conf/overrides/pets_cartpole.yaml sha256=Dvux5S861_G9lue7AA8iNyGXthWNFYYD0KP-RB1459g 378
mbrl/examples/conf/overrides/pets_cartpole_paper_version.yaml sha256=9lI289w209LRXCCIsIHDNhlqCxNbeDvFLhTPfYkRlwg 637
mbrl/examples/conf/overrides/pets_halfcheetah.yaml sha256=kIZJGKxdcALE0-5H3YK27Y_ycyD4HAg0QNq2IxyDlUY 503
mbrl/examples/conf/overrides/pets_hopper.yaml sha256=U64yVH8TCH4I5Fj6XMuUfSo8P1VqwpnZoVDzrpxcAjc 392
mbrl/examples/conf/overrides/pets_icem_cartpole.yaml sha256=d7fgyJ9YA3RWszLIYSPcukqcg5I2KlO-QP0Z7BLTk68 441
mbrl/examples/conf/overrides/pets_inv_pendulum.yaml sha256=_lv1IWiLwzRHUcOaN-aq9epV29mFwC8Z1ltbLzv01Mo 428
mbrl/examples/conf/overrides/pets_mppi_halfcheetah.yaml sha256=y_5Oh5n54Vt0feNx3ngKv8vFh-Dc9oyKYHmHpmRPb08 481
mbrl/examples/conf/overrides/pets_pusher.yaml sha256=8Tm2iYUTt6FV1A_kXSWjQ-eMyZJ5ZHFY5DF7UvrdAok 395
mbrl/examples/conf/overrides/pets_reacher.yaml sha256=eR0T7Mdo8T3EDixtSXYSg3ZqTfFFBRzO3wjnbKLv1s4 396
mbrl/examples/conf/overrides/planet_cartpole_balance.yaml sha256=s0VXpGipAaWX2Ss5q4PPOWm3oupezNV1XruYqcJGBlo 688
mbrl/examples/conf/overrides/planet_cartpole_swingup.yaml sha256=l3M3n-WauQvIJINIpooYvqV5RYgGbX4wf5NOokoi9B0 688
mbrl/examples/conf/overrides/planet_cheetah_run.yaml sha256=YFH0CBySpyzKArfH4Hw8B8cdFKPr-AT6X5mPe8vmjx0 678
mbrl/examples/conf/overrides/planet_cup_catch.yaml sha256=diy5yFiBCnUWmuEE75DhjSC9gxzstautIXnH8M5NXYo 690
mbrl/examples/conf/overrides/planet_finger_spin.yaml sha256=csoknHaltG_w-u_pjYD49JacLnYPO7YkNlt0tfq2Ek8 678
mbrl/examples/conf/overrides/planet_walker_walk.yaml sha256=kxIQPxhLFig77O56ol7ce6imn-r-Gp0_rbxD2aGadhY 678
mbrl/models/__init__.py sha256=ISJ5hjauTBYLB6WRBYlbU6wdjEtMuOpiP8FDk23-HmU 585
mbrl/models/basic_ensemble.py sha256=7tv4X3opWoXDfEPLUgV6M3FIStpq2caNNEy3zuYc7xM 11262
mbrl/models/gaussian_mlp.py sha256=k9SsU6gLrXecVCmONMwA_N1wL-4YH8w1egp0LSk8sfY 18351
mbrl/models/model.py sha256=rcfhaPVI462ypkeZF3mVA4EdEczIVuPqbYBh_gAEiIs 18507
mbrl/models/model_env.py sha256=i5RsGa2_jvn3diUIBWUovwUoDtG0XwsZRHFn0RB6OeI 8294
mbrl/models/model_trainer.py sha256=ONtBT7RecO8LJEyNPKmgHsJ44a6RSn4BqIT8IgNCQhY 12245
mbrl/models/one_dim_tr_model.py sha256=E8pzU-e5ZvvpBq7A1RSVxB5rbdK71aqNbPCCsAVwlEQ 13986
mbrl/models/planet.py sha256=7pbDCzKpDoXy4Oo5Gd2kB71UB2rW9YZvw_zuH0KnAzM 27455
mbrl/models/util.py sha256=lkA5kEke08tqI8sYoormMnqKoIGK3U4FjUbJIgKYsUQ 8195
mbrl/planning/__init__.py sha256=z-j2S5dXII7LlwtvhfpK0jqEvnSeUO8SWK0uCCGDShA 493
mbrl/planning/core.py sha256=K_IaJ_z-HAy3N5g5n_TgkedMs5ZYqhJca5rYebVDd6g 5468
mbrl/planning/linear_feedback.py sha256=przeldup3c1zk318vD03iGCQqmFZFGJ0XBYL8-0ZnXg 4055
mbrl/planning/sac_wrapper.py sha256=LPU7HwQohUFZtJsR6QtkSAY-xAYzWj9zLWP60elailY 1461
mbrl/planning/trajectory_opt.py sha256=3JRx3ihozMK_BB2FiRzSDKOfoplCOwEUUfFKNflT12o 31404
mbrl/third_party/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
mbrl/third_party/dmc2gym/__init__.py sha256=o1WKew5PsnHh_TXEF2YV8Ifd230q_5YCLqk9gE3OejU 1626
mbrl/third_party/dmc2gym/setup.py sha256=v0XyI4zYtQOtgSTxvxxMIy1fkxnJjTga9th9uZ8jZdY 348
mbrl/third_party/dmc2gym/wrappers.py sha256=esQlgHt04aZFXpGNxcMIbcQDccUPSZCT7vjpvpkETEI 5634
mbrl/third_party/pytorch_sac/__init__.py sha256=pJ7r_K_SnDMJ5MAriYgEgLxwogIkewmtCkEXtk47Hfs 157
mbrl/third_party/pytorch_sac/logger.py sha256=J2RHi-jSbojO7HqSO8OT2w2Pzgk3qperbHhzcnDXm_w 5974
mbrl/third_party/pytorch_sac/replay_buffer.py sha256=7KzqmNQTxzI5ZJUQ-U--dKkNseyVnveFHx85hF617eI 3495
mbrl/third_party/pytorch_sac/setup.py sha256=ycL3BbCwx87xSFSSYhkeZtsrVwQ1qDYkNUcq90gQHhw 551
mbrl/third_party/pytorch_sac/train.py sha256=OFgC45YmJaraMfhk6eYTsxuxpmMN5HLjw1TRbRkbnOs 4665
mbrl/third_party/pytorch_sac/utils.py sha256=-dIpwoscSBhHlDebrNcE59kHtfmJAM4Wbz4A-Ol2xYE 3639
mbrl/third_party/pytorch_sac/video.py sha256=yEHXZTrR6awJtBUqEGBRxXONvgRYvjxrWC8bhyyKdC4 982
mbrl/third_party/pytorch_sac/agent/__init__.py sha256=ajRgkaTcIsIykTuHgS0uWTwH08kRuzUqyK1SKHlUdv4 571
mbrl/third_party/pytorch_sac/agent/actor.py sha256=Q0zO9G-ypLqbaf2IT4x5Cr7QLP_ZZwUc8VZkI2mExDk 2814
mbrl/third_party/pytorch_sac/agent/critic.py sha256=8Etm3htMAlnfAmv_lF-m5oiAcMw79X1o3AeKRelnnZs 1254
mbrl/third_party/pytorch_sac/agent/sac.py sha256=Hg8eNcFUhnGvPS8-lA3XaCCqaHVo7FhR8_rpURXWsIQ 5767
mbrl/third_party/pytorch_sac_pranz24/__init__.py sha256=PUD-alISC97Cp0VFuTfi2XXi4yPSkmladAsD7z6ypEA 61
mbrl/third_party/pytorch_sac_pranz24/model.py sha256=zrqsvZOoTs4MeF6PDAYVcia1eA3CpqfP8RnSU8jceg8 5150
mbrl/third_party/pytorch_sac_pranz24/replay_memory.py sha256=RiOTDjABQH5yKC2PaJG6yj96iptFvdUnQzJ_MSVyi4w 1704
mbrl/third_party/pytorch_sac_pranz24/sac.py sha256=fapdN1rwrLSpvl8y0w1wC0_mStRzSiKtvssBIfbnAXI 8379
mbrl/third_party/pytorch_sac_pranz24/utils.py sha256=bnZ_ciMIb105TY44uCehXt5MIn_23Wm1SDDHCYLerx8 970
mbrl/util/__init__.py sha256=NqtH8Fptd09LV2k1mWgmzV4vsDhAQpFYftpls94KtcE 4207
mbrl/util/common.py sha256=2uM4-uDYPjYKn6pH-PBoKho3DPgaK1tCEGFNx7ErAgc 25339
mbrl/util/dmcontrol.py sha256=DByyjlYWTtmrVWFVRpYllm82ogoQY6vtecSefouy0VI 3783
mbrl/util/env.py sha256=es06ElJTD_Mxv31aYo1UDQRcwiuv8lYrJg0das889NY 11832
mbrl/util/logger.py sha256=2kqYEZAhKWZe5YyTiKh036NgM8uTy8Ul0n9H0mPS98E 8619
mbrl/util/math.py sha256=_Nki_TcVHXXrH4rnDg-y-S0gWZeAe3wFzivu7w4_eTM 15923
mbrl/util/mujoco.py sha256=qD5C9KQ5b7HRoVggBFm1KxXjOsAWPBYsuUTBvxkjv0g 4691
mbrl/util/pybullet.py sha256=NZcV9BRIpm_be9pW2QyEQ05QfNCBR4zNBsWrhjC0V3w 8794
mbrl/util/replay_buffer.py sha256=-FK44S9Ns_R5q9N-y5NgoFhwz8Yn-0CBwNOP4jTpVDU 28939
tests/__init__.py sha256=aRk5z30tsbGqr2IyqJjTQEXyoG0uueBnODd1kP5RoRY 198
tests/core/__init__.py sha256=aRk5z30tsbGqr2IyqJjTQEXyoG0uueBnODd1kP5RoRY 198
tests/core/test_common_utils.py sha256=_8gxjWzQe3xGZCH1cxof9J8QVr7RK5OR3SrpWKU-8p4 12706
tests/core/test_models.py sha256=m8Osiki0iXp6YAbMzjvrmpSjfQML42hEZamjQaxpJZQ 16621
tests/core/test_planning.py sha256=0ijFEyoYKSR3PcbTzFQsF4-Lw9bvgUeNGGGrCbhsAWY 2892
tests/core/test_replay_buffer.py sha256=AWATBjMuc3b_ybPeu5F1fwuGp1C-Y1EJM3zzkeRM870 25704
tests/dmcontrol/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/dmcontrol/test_util.py sha256=DK1fwbEHutwSmDbTj8sSdOd1FpFKYL0OFdX8CeZ3ZmA 2327
tests/mujoco/__init__.py sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU 0
tests/mujoco/test_diagnostics.py sha256=Usdkf_LdhP4O8m8ER0W8Z41F7irnVIpqVamjImJiWkM 5890
tests/mujoco/test_util.py sha256=iTmFD5LBwWnltIkKdnqQkRyvd2JN963YjaIScqRrBQ8 2603
mbrl-0.2.0.dist-info/LICENSE sha256=UkEte8fOQVfqYou6rLiCngqcs8WPV_mRdhJryM8r_IU 1086
mbrl-0.2.0.dist-info/METADATA sha256=rgvqib-Zim7xPP1tGXiROqisqvue4AFWZWZS6cZhM0U 14076
mbrl-0.2.0.dist-info/WHEEL sha256=pkctZYzUS4AYVn6dJ-7367OJZivF2e8RA9b_ZBjif18 92
mbrl-0.2.0.dist-info/top_level.txt sha256=_qvbQFQku6Og1bmW-r638SxSTcH6GREyBIdhEJbTy1Q 11
mbrl-0.2.0.dist-info/RECORD

top_level.txt

mbrl
tests