Skip to content

Commit

Permalink
Scope down sncosmo dependency
Browse files Browse the repository at this point in the history
  • Loading branch information
jeremykubica committed Jul 11, 2024
1 parent 9f99eed commit 03f1c82
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/tdastro/sources/sncomso_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
https://sncosmo.readthedocs.io/en/stable/models.html
"""

import sncosmo
from sncosmo.models import Model

from tdastro.base_models import PhysicalModel

Expand All @@ -30,7 +30,7 @@ class SncosmoModel(PhysicalModel):
def __init__(self, model_name, **kwargs):
super().__init__(**kwargs)
self.model_name = model_name
self.model = sncosmo.Model(source=model_name)
self.model = Model(source=model_name)

def __str__(self):
"""Return the string representation of the model."""
Expand Down

0 comments on commit 03f1c82

Please sign in to comment.