Skip to content

Commit 43ff998

Browse files
authored
devolearn: upgrade lineage population model (#38)
1 parent 61eb88d commit 43ff998

File tree

3 files changed

+2
-2
lines changed

3 files changed

+2
-2
lines changed

devolearn/lineage_population_model/lineage_population_model.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -36,8 +36,8 @@ def __init__(self, device = "cpu"):
3636
self.model_dir = os.path.dirname(__file__)
3737
self.scaler = joblib.load(self.model_dir + "/" + 'scaler/scaler.gz')
3838

39-
self.model_url = "https://github.com/DevoLearn/devolearn/raw/master/devolearn/lineage_population_model/estimate_lineage_population.pt"
40-
self.model_name = "estimate_lineage_population.pt"
39+
self.model_url = "https://github.com/DevoLearn/devolearn/raw/master/devolearn/lineage_population_model/estimate_lineage_population.pth"
40+
self.model_name = "estimate_lineage_population.pth"
4141
# print("at : ", os.path.dirname(__file__))
4242

4343

-5 Bytes
Binary file not shown.

0 commit comments

Comments
 (0)