Moved hdf functionality to anesthetic.read

handley-lab · Jun 29, 2023 · f432f69 · f432f69
1 parent db6bc25
commit f432f69
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 5 deletions.
diff --git a/anesthetic/__init__.py b/anesthetic/__init__.py
@@ -2,7 +2,7 @@
 import anesthetic.samples
 import anesthetic.plot
 import anesthetic.read.chain
-import anesthetic.io
+import anesthetic.read.hdf
 
 import pandas
 import pandas.plotting._core
@@ -48,5 +48,5 @@ def wrapper(backend=None):
 make_2d_axes = anesthetic.plot.make_2d_axes
 make_1d_axes = anesthetic.plot.make_1d_axes
 
-read_hdf = anesthetic.io.read_hdf
+read_hdf = anesthetic.read.hdf.read_hdf
 read_chains = anesthetic.read.chain.read_chains
diff --git a/anesthetic/io.py → anesthetic/read/hdf.py b/anesthetic/io.py → anesthetic/read/hdf.py
diff --git a/anesthetic/samples.py b/anesthetic/samples.py
@@ -470,8 +470,9 @@ def tex(self):
             )
 
     def to_hdf(self, path_or_buf, key, *args, **kwargs):  # noqa: D102
-        import anesthetic.io
-        anesthetic.io.to_hdf(path_or_buf, key, self, *args, **kwargs)
+        import anesthetic.read.hdf
+        return anesthetic.read.hdf.to_hdf(path_or_buf, key, self,
+                                          *args, **kwargs)
 
 
 class MCMCSamples(Samples):

diff --git a/tests/test_reader.py b/tests/test_reader.py
@@ -12,7 +12,7 @@
 from anesthetic.read.cobaya import read_cobaya
 from anesthetic.read.multinest import read_multinest
 import pandas._testing as tm
-from anesthetic.io import HDFStore, read_hdf
+from anesthetic.read.hdf import HDFStore, read_hdf
 try:
     import getdist
 except ImportError: