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

Merge branch 'lukas_issue070_bug_wrong-model-plot-path' into 'develop'

apply bugfix to dev branch too

See merge request toar/machinelearningtools!65
parents 164d1cdb 7997cf3c
No related branches found
No related tags found
3 merge requests!90WIP: new release update,!89Resolve "release branch / CI on gpu",!65apply bugfix to dev branch too
Pipeline #31472 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