fix: automatically append .h5 suffix when rating datasets

AFM-analysis · Jun 15, 2021 · a8cf6cf · a8cf6cf
1 parent 3346e24
commit a8cf6cf
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/CHANGELOG b/CHANGELOG
@@ -1,6 +1,7 @@
 0.9.0
  - feat: allow users to enter missing metadata during loading
    of data files (#5)
+ - fix: automatically append .h5 suffix when rating datasets
  - ref: directory for importing training datasets for rating changed
    (but I think this currently affects nobody)
  - setup: dump appdirs dependency (moved to QSettings)

diff --git a/pyjibe/fd/rating_iface.py b/pyjibe/fd/rating_iface.py
@@ -1,4 +1,5 @@
 import os
+import pathlib
 import pkg_resources
 
 from nanite.rate import io as nio
@@ -17,6 +18,9 @@ def __init__(self, fdui, path):
         UiUserRatingBase.__init__(self)
         self.setupUi(self)
         self.fdui = fdui
+        path = pathlib.Path(path)
+        if not path.suffix == ".h5":
+            path = path.with_name(path.name + ".h5")
         self.path = path
 
         self.initial_stuff()