Skip to content

Commit

Permalink
amend
Browse files Browse the repository at this point in the history
Signed-off-by: Matteo Bettini <[email protected]>
  • Loading branch information
matteobettini committed Sep 11, 2023
1 parent 00fafa3 commit 5379914
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion benchmarl/models/mlp.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,9 +5,9 @@
from tensordict import TensorDictBase
from torch import nn
from torchrl.modules import MLP, MultiAgentMLP
from utils import read_yaml_config

from benchmarl.models.common import Model, ModelConfig, parse_model_config
from benchmarl.utils import read_yaml_config


class Mlp(Model):
Expand Down

0 comments on commit 5379914

Please sign in to comment.