amend

Signed-off-by: Matteo Bettini <[email protected]>

benchmarl/algorithms/iddpg.py

@@ -1,8 +1,7 @@
 from dataclasses import dataclass, MISSING
-from typing import Dict, Optional, Type
+from typing import Dict, Optional, Tuple, Type
 
 import torch
-from black import Tuple
 from tensordict import TensorDictBase
 from tensordict.nn import TensorDictModule, TensorDictSequential
 from torchrl.data import (

benchmarl/algorithms/ippo.py

@@ -1,8 +1,7 @@
 from dataclasses import dataclass, MISSING
-from typing import Dict, Optional, Type
+from typing import Dict, Optional, Tuple, Type
 
 import torch
-from black import Tuple
 from tensordict import TensorDictBase
 from tensordict.nn import TensorDictModule, TensorDictSequential
 from tensordict.nn.distributions import NormalParamExtractor

benchmarl/algorithms/iql.py

@@ -1,8 +1,7 @@
 from dataclasses import dataclass, MISSING
-from typing import Dict, Optional, Type
+from typing import Dict, Optional, Type, Tuple
 
 import torch
-from black import Tuple
 from tensordict import TensorDictBase
 from tensordict.nn import TensorDictModule, TensorDictSequential
 from torchrl.data import (

benchmarl/algorithms/isac.py

@@ -1,8 +1,7 @@
 from dataclasses import dataclass, MISSING
-from typing import Dict, Optional, Type, Union
+from typing import Dict, Optional, Tuple, Type, Union
 
 import torch
-from black import Tuple
 from tensordict import TensorDictBase
 from tensordict.nn import NormalParamExtractor, TensorDictModule, TensorDictSequential
 from torch.distributions import Categorical

benchmarl/algorithms/maddpg.py

@@ -1,8 +1,7 @@
 from dataclasses import dataclass, MISSING
-from typing import Dict, Optional, Type
+from typing import Dict, Optional, Tuple, Type
 
 import torch
-from black import Tuple
 from tensordict import TensorDictBase
 from tensordict.nn import TensorDictModule, TensorDictSequential
 from torchrl.data import (

benchmarl/algorithms/mappo.py

@@ -1,8 +1,7 @@
 from dataclasses import dataclass, MISSING
-from typing import Dict, Optional, Type
+from typing import Dict, Optional, Tuple, Type
 
 import torch
-from black import Tuple
 from tensordict import TensorDictBase
 from tensordict.nn import TensorDictModule, TensorDictSequential
 from tensordict.nn.distributions import NormalParamExtractor

benchmarl/algorithms/masac.py

@@ -1,8 +1,7 @@
 from dataclasses import dataclass, MISSING
-from typing import Dict, Optional, Type, Union
+from typing import Dict, Optional, Tuple, Type, Union
 
 import torch
-from black import Tuple
 from tensordict import TensorDictBase
 from tensordict.nn import NormalParamExtractor, TensorDictModule, TensorDictSequential
 from torch.distributions import Categorical

benchmarl/algorithms/qmix.py

@@ -1,8 +1,7 @@
 from dataclasses import dataclass, MISSING
-from typing import Dict, Optional, Type
+from typing import Dict, Optional, Tuple, Type
 
 import torch
-from black import Tuple
 from tensordict import TensorDictBase
 from tensordict.nn import TensorDictModule, TensorDictSequential
 from torchrl.data import (

benchmarl/algorithms/vdn.py

@@ -1,8 +1,7 @@
 from dataclasses import dataclass, MISSING
-from typing import Dict, Optional, Type
+from typing import Dict, Optional, Tuple, Type
 
 import torch
-from black import Tuple
 from tensordict import TensorDictBase
 from tensordict.nn import TensorDictModule, TensorDictSequential
 from torchrl.data import (

examples/simple_hydra_run.py

@@ -2,7 +2,6 @@
 
 from benchmarl.environments import task_config_registry
 from benchmarl.experiment import Experiment
-
 from benchmarl.hydra_run import load_model_from_hydra_config
 from hydra.core.hydra_config import HydraConfig
 from omegaconf import DictConfig, OmegaConf