moved fktable_hasher to n3fit_checks_provider

Author: jekoorn

n3fit/src/n3fit/checks.py

@@ -2,7 +2,6 @@
 This module contains checks to be perform by n3fit on the input
 """
 
-import hashlib
 import logging
 import numbers
 import os
@@ -28,7 +27,6 @@
     "layer_type",
     "dropout",
 ]
-MD5FK_FILENAME = "md5fk"
 
 
 def _is_floatable(num):
@@ -566,25 +564,3 @@ def check_eko_exists(theoryid):
         _ = FallbackLoader().check_eko(theoryid.id)
     except FileNotFoundError:
         log.error(f"No eko found for {theoryid}")
-
-
-def fktable_hasher(data, output_path):
-    """Writes a hash of the fk-tables to a log file.
-    This hash can be used to ensure whether two
-    (supposedly identical) fk-tables of the same
-    theory and dataset are numerically identical.
-    """
-    md5fk_path = output_path / MD5FK_FILENAME
-    # Open a file to write in
-    with open(md5fk_path, "w") as f:
-        # Loop through the dataspecs object
-        for dataset in data.datasets:
-            fkspecs = dataset.fkspecs
-            for fk in fkspecs:
-                # Make a list of the FK tables
-                table_names = [name for group in fk.metadata.FK_tables for name in group]
-                for fkpath, table_name in zip(fk.fkpath, table_names):
-                    for fkpath, table_name in zip(fk.fkpath, table_names):
-                        fkhash = hashlib.md5(fkpath.read_bytes()).hexdigest()
-                        f.write(f"{table_name} {fkhash}\n")
-    log.info(f"FK-table hash written to {md5fk_path}")

n3fit/src/n3fit/n3fit_checks_provider.py

@@ -5,6 +5,7 @@
 """
 
 import n3fit.checks
+import hashlib
 
 
 @n3fit.checks.check_consistent_basis
@@ -41,5 +42,24 @@ def n3fit_checks_action(
 def evolven3fit_checks_action(theoryid):
     return
 
+
 def fktable_hasher(data, output_path):
-    return
+    """Writes a hash of the fk-tables to a log file.
+    This hash can be used to ensure whether two
+    (supposedly identical) fk-tables of the same
+    theory and dataset are numerically identical.
+    """
+    MD5FK_FILENAME = "md5fk"
+    md5fk_path = output_path / MD5FK_FILENAME
+    # Open a file to write in
+    with open(md5fk_path, "w") as f:
+        # Loop through the dataspecs object
+        for dataset in data.datasets:
+            fkspecs = dataset.fkspecs
+            for fk in fkspecs:
+                # Make a list of the FK tables
+                table_names = [name for group in fk.metadata.FK_tables for name in group]
+                for fkpath, table_name in zip(fk.fkpath, table_names):
+                    for fkpath, table_name in zip(fk.fkpath, table_names):
+                        fkhash = hashlib.md5(fkpath.read_bytes()).hexdigest()
+                        f.write(f"{table_name} {fkhash}\n")