Typing for python=3.8

Author: PatReis

kgcnn/data/base.py

@@ -3,7 +3,7 @@
 import tensorflow as tf
 import pandas as pd
 import os
-from typing import Union
+from typing import Union, List
 from collections.abc import MutableMapping
 from kgcnn.data.utils import save_pickle_file, load_pickle_file, ragged_tensor_from_nested_numpy
 from kgcnn.graph.base import GraphNumpyContainer, GraphDict
@@ -493,7 +493,7 @@ def message_error(msg):
                 message_error("Can not check shape for '%s'." % x["name"])
         return
 
-    def set_methods(self, method_list: list[dict]) -> None:
+    def set_methods(self, method_list: List[dict]) -> None:
         r"""Apply a list of serialized class-methods on the dataset.
 
         This can extend the config-serialization scheme in :obj:`kgcnn.utils.serial`.

kgcnn/io/loader.py

@@ -1,6 +1,6 @@
 import numpy as np
 import tensorflow as tf
-from typing import Union
+from typing import Union, List
 from kgcnn.data.utils import ragged_tensor_from_nested_numpy
 from kgcnn.data.base import MemoryGraphDataset
 ks = tf.keras
@@ -10,9 +10,9 @@ class GraphBatchLoader(ks.utils.Sequence):
     r"""Example (minimal) implementation of a graph batch loader based on :obj:`ks.utils.Sequence`."""
 
     def __init__(self,
-                 data: Union[list[dict], MemoryGraphDataset],
-                 inputs: Union[dict, list[dict]],
-                 outputs: Union[dict, list[dict]],
+                 data: Union[List[dict], MemoryGraphDataset],
+                 inputs: Union[dict, List[dict]],
+                 outputs: Union[dict, List[dict]],
                  batch_size: int = 32,
                  shuffle: bool = False):
         """Initialization with data and input information.
@@ -65,7 +65,7 @@ def on_epoch_end(self):
         self._shuffle_indices()
 
     @staticmethod
-    def _to_tensor(item, is_ragged):
+    def _to_tensor(item: Union[np.ndarray, list], is_ragged: bool):
         if is_ragged:
             return ragged_tensor_from_nested_numpy(item)
         else: