From 56ea22bf4a3d058b4df79c7be119a3fb1dd40fad Mon Sep 17 00:00:00 2001
From: FBurkhardt <fburkhardt@audeering.com>
Date: Tue, 15 Oct 2024 12:53:08 +0200
Subject: [PATCH] 0.90.4

---
 CHANGELOG.md           | 4 ++++
 nkululeko/constants.py | 2 +-
 nkululeko/multidb.py   | 6 +++++-
 3 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index b72945b..44b3f3b 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -1,6 +1,10 @@
 Changelog
 =========
 
+Version 0.90.4
+--------------
+* added plot format for multidb
+
 Version 0.90.3
 --------------
 * refactorings and documentations
diff --git a/nkululeko/constants.py b/nkululeko/constants.py
index fd6b826..0043fb3 100644
--- a/nkululeko/constants.py
+++ b/nkululeko/constants.py
@@ -1,2 +1,2 @@
-VERSION="0.90.3"
+VERSION="0.90.4"
 SAMPLING_RATE = 16000
diff --git a/nkululeko/multidb.py b/nkululeko/multidb.py
index a7c9cb6..39391c3 100644
--- a/nkululeko/multidb.py
+++ b/nkululeko/multidb.py
@@ -115,7 +115,11 @@ def main(src_dir):
     print(repr(results))
     print(repr(last_epochs))
     root = os.path.join(config["EXP"]["root"], "")
-    plot_name = f"{root}/heatmap.png"
+    try:
+        format = config["PLOT"]["format"]
+        plot_name = f"{root}/heatmap.{format}"
+    except KeyError:
+        plot_name = f"{root}/heatmap.png"
     plot_heatmap(results, last_epochs, datasets, plot_name, config, datasets)