Skip to content
Snippets Groups Projects
Commit 7997cf3c authored by lukas leufen's avatar lukas leufen
Browse files

replaced split by rsplit, bug fixed

parent 3725c58e
No related branches found
No related tags found
2 merge requests!65apply bugfix to dev branch too,!64Resolve "BUG: wrong model plot path"
Pipeline #31468 passed
......@@ -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)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment