Merge pull request #4 from microsoft/dev/mallamanis/expose-ifaces

Expose `load_metadata_and_create_network`
microsoft · Sep 28, 2020 · 2cb7ed6 · 2cb7ed6
2 parents 1f1001f + 5eeedfb
commit 2cb7ed6
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/ptgnn/baseneuralmodel/trainer.py b/ptgnn/baseneuralmodel/trainer.py
@@ -108,6 +108,13 @@ def neural_module(self) -> TNeuralModule:
     def neural_module(self, nn: TNeuralModule):
         self.__neural_network = nn
 
+    def load_metadata_and_create_network(
+        self, training_data: Iterable[TRawDatapoint], parallelize: bool, show_progress_bar: bool
+    ) -> None:
+        return self.__load_metadata_and_create_network(
+            training_data, parallelize, show_progress_bar
+        )
+
     def __load_metadata_and_create_network(
         self, training_data: Iterable[TRawDatapoint], parallelize: bool, show_progress_bar: bool
     ) -> None:
@@ -134,7 +141,8 @@ def __load_metadata_and_create_network(
             m(self.__model)
 
         self.LOGGER.info(
-            "Model Definition:\n %s", json.dumps(dict(self.__model.model_definition), indent=2),
+            "Model Definition:\n %s",
+            json.dumps(dict(self.__model.model_definition), indent=2),
         )
 
     def __save_checkpoint(self) -> None: