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
Branches
Tags
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