Remove manual calls to datamodule methods

Author: SpirinEgor

code2seq/code2class_wrapper.py

@@ -39,8 +39,6 @@ def test_code2class(config: DictConfig):
 
     # Load data module
     data_module = PathContextDataModule(config.data_folder, config.data)
-    data_module.prepare_data()
-    data_module.setup()
 
     # Load model
     code2class = Code2Class.load_from_checkpoint(config.checkpoint, map_location=torch.device("cpu"))

code2seq/code2seq_wrapper.py

@@ -39,8 +39,6 @@ def test_code2seq(config: DictConfig):
 
     # Load data module
     data_module = PathContextDataModule(config.data_folder, config.data)
-    data_module.prepare_data()
-    data_module.setup()
 
     # Load model
     code2seq = Code2Seq.load_from_checkpoint(config.checkpoint, map_location=torch.device("cpu"))

code2seq/typed_code2seq_wrapper.py

@@ -39,8 +39,6 @@ def test_typed_code2seq(config: DictConfig):
 
     # Load data module
     data_module = TypedPathContextDataModule(config.data_folder, config.data)
-    data_module.prepare_data()
-    data_module.setup()
 
     # Load model
     typed_code2seq = TypedCode2Seq.load_from_checkpoint(config.checkpoint, map_location=torch.device("cpu"))