fix mypy errors

Author: smanolloff

src/imitation/algorithms/adversarial/common.py

@@ -2,7 +2,7 @@
 import abc
 import dataclasses
 import logging
-from typing import Iterable, Iterator, Mapping, Optional, Type, overload
+from typing import Iterable, Iterator, Mapping, Optional, Type, List, overload
 
 import numpy as np
 import torch as th
@@ -408,7 +408,10 @@ def train_gen(
         if learn_kwargs is None:
             learn_kwargs = {}
 
-        callbacks = [self.gen_callback]
+        callbacks: List[BaseCallback] = []
+
+        if self.gen_callback:
+            callbacks.append(self.gen_callback)
 
         if isinstance(callback, list):
             callbacks.extend(callback)

src/imitation/scripts/train_adversarial.py

@@ -33,7 +33,7 @@ def __init__(
         interval: int,
     ):
         """Creates new Checkpoint callback."""
-        super().__init__(self)
+        super().__init__()
         self.trainer = trainer
         self.log_dir = log_dir
         self.interval = interval