From 298ed79e51c21019bcd187cc977681002656e11c Mon Sep 17 00:00:00 2001 From: mounika-2604 Date: Mon, 5 Oct 2026 23:32:48 +0530 Subject: [PATCH] Move Fermi-Hubbard publication data fetching --- recirq/fermi_hubbard/__init__.py | 5 +++++ recirq/fermi_hubbard/publication.py | 2 ++ recirq/fermi_hubbard/publication_test.py | 12 ++++++++++++ 3 files changed, 19 insertions(+) diff --git a/recirq/fermi_hubbard/__init__.py b/recirq/fermi_hubbard/__init__.py index 95a2ecbc..153909c5 100644 --- a/recirq/fermi_hubbard/__init__.py +++ b/recirq/fermi_hubbard/__init__.py @@ -76,3 +76,8 @@ apply_rescalings_to_bundles, find_bundles_rescalings ) + +from recirq.fermi_hubbard.publication import ( + fetch_publication_data, + get_publication_data, +) diff --git a/recirq/fermi_hubbard/publication.py b/recirq/fermi_hubbard/publication.py index be779db5..06187686 100644 --- a/recirq/fermi_hubbard/publication.py +++ b/recirq/fermi_hubbard/publication.py @@ -282,3 +282,5 @@ def fetch_publication_data( zfile.extractall(base_dir) print(f"Successfully downloaded and extracted {file_name}.\n") + +get_publication_data = fetch_publication_data diff --git a/recirq/fermi_hubbard/publication_test.py b/recirq/fermi_hubbard/publication_test.py index a3c87baf..3c3803e7 100644 --- a/recirq/fermi_hubbard/publication_test.py +++ b/recirq/fermi_hubbard/publication_test.py @@ -29,3 +29,15 @@ def try_fetch_publication_data(): fetch_publication_data(base_dir=base_dir) assert os.path.exists(base_dir + os.path.sep + "trapping_3u3d") + + +def test_publication_data_functions(): + import recirq.fermi_hubbard as fh + from recirq.fermi_hubbard.publication import ( + fetch_publication_data, + get_publication_data, + ) + + assert get_publication_data is fetch_publication_data + assert fh.fetch_publication_data is fetch_publication_data + assert fh.get_publication_data is fetch_publication_data