diff --git a/notebooks/thrifty_demo.ipynb b/notebooks/thrifty_demo.ipynb index 61eec492..27b7c8a2 100644 --- a/notebooks/thrifty_demo.ipynb +++ b/notebooks/thrifty_demo.ipynb @@ -24,7 +24,7 @@ "\n", "To get a pretrained crepe, just ask for it by name. 🍰!\n", "\n", - "Here we get the `ThriftyHumV1.0-45` model:" + "Here we get the `ThriftyHumV0.2-45` model:" ] }, { diff --git a/tests/conftest.py b/tests/conftest.py index 9b7b8372..c88350cb 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -13,6 +13,6 @@ def pcp_df(): ) df = add_shm_model_outputs_to_pcp_df( df, - pretrained.load("ThriftyHumV1.0-45"), + pretrained.load("ThriftyHumV0.2-45"), ) return df diff --git a/tests/test_multihit.py b/tests/test_multihit.py index 4037cc98..aebc8c53 100644 --- a/tests/test_multihit.py +++ b/tests/test_multihit.py @@ -64,7 +64,7 @@ @pytest.fixture def mini_multihit_train_val_datasets(): df = pd.read_csv("data/wyatt-10x-1p5m_pcp_2023-11-30_NI.first100.csv.gz") - crepe = pretrained.load("ThriftyHumV1.0-45") + crepe = pretrained.load("ThriftyHumV0.2-45") df = multihit.prepare_pcp_df(df, crepe, 500) return multihit.train_test_datasets_of_pcp_df(df)