possible modification of typing

Limmen · Aug 18, 2023 · fa90933 · fa90933
1 parent c562e4c
commit fa90933
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/...tion-system/libs/csle-common/src/csle_common/dao/system_identification/gp_system_model.py b/...tion-system/libs/csle-common/src/csle_common/dao/system_identification/gp_system_model.py
@@ -34,7 +34,7 @@ def __init__(self, emulation_env_name: str, emulation_statistic_id: int,
         self.emulation_env_name = emulation_env_name
         self.emulation_statistic_id = emulation_statistic_id
         self.id = -1
-        self.conditionals_kl_divergences: Dict[str, Dict[str, Dict[str, Union[str, float]]]] = {}
+        self.conditionals_kl_divergences: Dict[str, Dict[str, Dict[str, float]]] = {}
         self.compute_kl_divergences()
 
     def compute_kl_divergences(self) -> None:
@@ -58,7 +58,7 @@ def compute_kl_divergences(self) -> None:
                                 metric_distributions_condition_2[0].conditional_name][metric_dist.metric_name]):
                         self.conditionals_kl_divergences[metric_distributions_condition_1[0].conditional_name][
                             metric_distributions_condition_2[0].conditional_name][
-                            metric_dist.metric_name] = "inf"
+                            metric_dist.metric_name] = math.inf
 
     @staticmethod
     def from_dict(d: Dict[str, Any]) -> "GPSystemModel":

diff --git a/simulation-system/libs/csle-common/src/csle_common/dao/training/hparam.py b/simulation-system/libs/csle-common/src/csle_common/dao/training/hparam.py
@@ -7,7 +7,7 @@ class HParam(JSONSerializable):
     DTO class representing a hyperparameter
     """
 
-    def __init__(self, value: Union[int, float, str, List], name: str, descr: str):
+    def __init__(self, value: Union[int, float, str, List[Any]], name: str, descr: str):
         """
         Initializes the DTO
 

diff --git a/simulation-system/libs/csle-common/src/csle_common/models/fnn_w_softmax.py b/simulation-system/libs/csle-common/src/csle_common/models/fnn_w_softmax.py
@@ -2,7 +2,7 @@
 A FNN model with Softmax output defined in PyTorch
 """
 import torch
-
+from typing import Union, Any, List
 
 class FNNwithSoftmax(torch.nn.Module):
     """
@@ -11,8 +11,8 @@ class FNNwithSoftmax(torch.nn.Module):
     Sub-classing the torch.nn.Module to be able to use high-level API for creating the custom network
     """
 
-    def __init__(self, input_dim: int, output_dim: int, hidden_dim: int, num_hidden_layers: int = 2,
-                 hidden_activation: str = "ReLU"):
+    def __init__(self, input_dim: int, output_dim: int, hidden_dim: Union[int, float, str, List[Any]],
+                 num_hidden_layers: int = 2, hidden_activation: str = "ReLU"):
         """
         Builds the model