Commit a2686526 authored by Nathan Frey's avatar Nathan Frey
Browse files

Fix type annotation inconsistencies

parent 7f5297b6
Loading
Loading
Loading
Loading
+3 −3
Original line number Diff line number Diff line
@@ -45,9 +45,9 @@ def load_mydataset(
    reload: bool = True,
    data_dir: Optional[str] = None,
    save_dir: Optional[str] = None,
    featurizer_kwargs: Optional[Dict[str, object]] = {},
    splitter_kwargs: Optional[Dict[str, object]] = {},
    transformer_kwargs: Optional[Dict[str, Dict[str, object]]] = {},
    featurizer_kwargs: Dict[str, object] = {},
    splitter_kwargs: Dict[str, object] = {},
    transformer_kwargs: Dict[str, Dict[str, object]] = {},
    **kwargs) -> Tuple[List, Tuple, List]:
  """Load mydataset.

+2 −2
Original line number Diff line number Diff line
@@ -83,9 +83,9 @@ def load_bandgap(
    Path to datasets.
  save_dir : str, optional
    Path to featurized datasets.
  featurizer_kwargs : Optional[Dict[str, Any]]
  featurizer_kwargs : Dict[str, Any]
    Specify parameters to featurizer, e.g. {"size": 1024}
  splitter_kwargs : Optional[Dict[str, Any]]
  splitter_kwargs : Dict[str, Any]
    Specify parameters to splitter, e.g. {"seed": 42}
  transformer_kwargs : dict
    Maps transformer names to constructor arguments, e.g.
+2 −2
Original line number Diff line number Diff line
@@ -81,9 +81,9 @@ def load_perovskite(
    Path to datasets.
  save_dir : str, optional
    Path to featurized datasets.
  featurizer_kwargs : Optional[Dict[str, Any]]
  featurizer_kwargs : Dict[str, Any]
    Specify parameters to featurizer, e.g. {"size": 1024}
  splitter_kwargs : Optional[Dict[str, Any]]
  splitter_kwargs : Dict[str, Any]
    Specify parameters to splitter, e.g. {"seed": 42}
  transformer_kwargs : dict
    Maps transformer names to constructor arguments, e.g.