fixing db.Session in dapdb schema

sdss · Apr 13, 2020 · 10e683b · 10e683b
1 parent 13b5c08
commit 10e683b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/sdssdb/sqlalchemy/mangadb/dapdb.py b/python/sdssdb/sqlalchemy/mangadb/dapdb.py
@@ -48,7 +48,7 @@ def ftype(self):
 
     @property
     def partner(self):
-        session = db.Session.object_session(self)
+        session = database.Session.object_session(self)
         return session.query(File).join(Structure, datadb.Cube, FileType).filter(
             Structure.pk == self.structure.pk, datadb.Cube.pk == self.cube.pk,
             FileType.pk != self.filetype.pk).one()
@@ -114,7 +114,7 @@ class Hdu(Base):
     def header(self):
         '''Returns an astropy header'''
 
-        session = db.Session.object_session(self)
+        session = database.Session.object_session(self)
         data = session.query(HeaderKeyword.name, HeaderValue.value,
                              HeaderValue.comment).join(HeaderValue, HduToHeaderValue).filter(
             HduToHeaderValue.header_value_pk == HeaderValue.pk,
@@ -280,7 +280,7 @@ def get3DCube(self, extension='flux'):
 
         """
 
-        session = db.Session.object_session(self)
+        session = database.Session.object_session(self)
         spaxels = session.query(getattr(ModelSpaxel, extension)).filter(
             ModelSpaxel.modelcube_pk == self.pk).order_by(ModelSpaxel.x, ModelSpaxel.y).all()