From 7997cf3c4f14a9339870bb5ace7dc2dda578fa34 Mon Sep 17 00:00:00 2001
From: lukas leufen <l.leufen@fz-juelich.de>
Date: Mon, 9 Mar 2020 17:16:11 +0100
Subject: [PATCH] replaced split by rsplit, bug fixed

---
 src/run_modules/model_setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/run_modules/model_setup.py b/src/run_modules/model_setup.py
index d2c8d93f..32ca0d2e 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)
-- 
GitLab