fix types

src/gluonts/torch/model/i_transformer/estimator.py

@@ -67,7 +67,7 @@ class ITransformerEstimator(PyTorchLightningEstimator):
  Number of time steps prior to prediction time that the model
  takes as inputs (default: ``10 * prediction_length``).
  d_model
- Size of hidden layers in the Transformer encoder.
+ Size of latent in the Transformer encoder.
  nhead
  Number of attention heads in the Transformer encoder which must divide d_model.
  dim_feedforward

src/gluonts/torch/model/i_transformer/module.py

@@ -11,7 +11,7 @@
 # express or implied. See the License for the specific language governing
 # permissions and limitations under the License.
 
-from typing import Tuple
+from typing import Optional, Tuple
 
 import torch
 from torch import nn
@@ -69,7 +69,7 @@ def __init__(
  activation: str,
  norm_first: bool,
  num_encoder_layers: int,
- scaling: str,
+ scaling: Optional[str],
  distr_output=StudentTOutput(),
  ) -> None:
  super().__init__()