diff --git a/n3fit/src/n3fit/io/writer.py b/n3fit/src/n3fit/io/writer.py index 1f43a3ca7a..e4ae14160e 100644 --- a/n3fit/src/n3fit/io/writer.py +++ b/n3fit/src/n3fit/io/writer.py @@ -10,6 +10,7 @@ import json import numpy as np from reportengine.compat import yaml +import n3fit from n3fit.msr import compute_arclength @@ -241,3 +242,7 @@ def storefit( # create .time file with open(f"{replica_path}/{fitname}.time", "w") as fs: json.dump(timings, fs, indent=2) + + # create .version file + with open(f"{replica_path}/version.info", "w") as fs: + fs.write(n3fit.__version__) diff --git a/n3fit/src/n3fit/tests/test_fit.py b/n3fit/src/n3fit/tests/test_fit.py index a650a1a042..379cc08bfb 100644 --- a/n3fit/src/n3fit/tests/test_fit.py +++ b/n3fit/src/n3fit/tests/test_fit.py @@ -24,6 +24,7 @@ from collections import namedtuple from numpy.testing import assert_almost_equal from reportengine.compat import yaml +import n3fit from n3fit.performfit import initialize_seeds log = logging.getLogger(__name__) @@ -103,6 +104,11 @@ def test_performfit(): fitting_time = times["walltime"]["replica_set_to_replica_fitted"] f.close() assert fitting_time < EXPECTED_MAX_FITTIME + version_path = tmp_path / f"{QUICKNAME}/nnfit/replica_{REPLICA}/version.info" + f = open(version_path, "r") + version = f.read() + f.close() + assert version == n3fit.__version__ def test_hyperopt():