Skip to content

[BUG] - MarioRL tutorial does not work on the collab #2627

Closed
@malfet

Description

@malfet

Add Link

https://pytorch.org/tutorials/intermediate/mario_rl_tutorial.html

Describe the bug

Click on Run in Google Collab button and Run all cells

It will fail while trying to import tensordict:

---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
[<ipython-input-3-8f359b2f0b63>](https://localhost:8080/#) in <cell line: 21>()
     19 import gym_super_mario_bros
     20 
---> 21 from tensordict import TensorDict
     22 from torchrl.data import TensorDictReplayBuffer, LazyMemmapStorage

ModuleNotFoundError: No module named 'tensordict'

This is a regression introduced by #2381

Describe your environment

GoogleCollab

cc @sekyondaMeta @svekars @carljparker @NicolasHug @kit1980 @subramen

Metadata

Metadata

Assignees

No one assigned

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions