diff --git a/src/run_modules/model_setup.py b/src/run_modules/model_setup.py index d2c8d93fc957ecb2990e99000cbd3588e2d83eef..32ca0d2e82af32d8164d80ac42731e10f431a458 100644 --- a/src/run_modules/model_setup.py +++ b/src/run_modules/model_setup.py @@ -100,5 +100,5 @@ class ModelSetup(RunEnvironment): def plot_model(self): # pragma: no cover with tf.device("/cpu:0"): - file_name = f"{self.model_name.split(sep='.')[0]}.pdf" + file_name = f"{self.model_name.rsplit('.', 1)[0]}.pdf" keras.utils.plot_model(self.model, to_file=file_name, show_shapes=True, show_layer_names=True)