From 9238fff0f50c8b740151364b5907c4fca017d381 Mon Sep 17 00:00:00 2001 From: Kashif Rasul Date: Thu, 18 Feb 2021 12:11:48 +0100 Subject: [PATCH] fixed typo --- pts/model/transformer/transformer_estimator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pts/model/transformer/transformer_estimator.py b/pts/model/transformer/transformer_estimator.py index 4a9d7c4..4adc188 100644 --- a/pts/model/transformer/transformer_estimator.py +++ b/pts/model/transformer/transformer_estimator.py @@ -251,7 +251,7 @@ class TransformerEstimator(PyTorchEstimator): copy_parameters(trained_network, prediction_network) input_names = get_module_forward_input_names(prediction_network) - prediction_splitter = self._create_instance_splitter("test") + prediction_splitter = self.create_instance_splitter("test") return PyTorchPredictor( input_transform=transformation + prediction_splitter,