Skip to content

Commit

Permalink
moving to 0.2
Browse files Browse the repository at this point in the history
  • Loading branch information
matsen committed Nov 13, 2024
1 parent 039d7be commit b33fe6a
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion notebooks/thrifty_demo.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -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:"
]
},
{
Expand Down
2 changes: 1 addition & 1 deletion tests/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
2 changes: 1 addition & 1 deletion tests/test_multihit.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)

Expand Down

0 comments on commit b33fe6a

Please sign in to comment.