Update element_session/export/nwb.py

Co-authored-by: Kabilar Gunalan <[email protected]>

Author: kushalbakshi

element_session/export/nwb.py

@@ -5,18 +5,13 @@
 
 from .. import session_with_id, session_with_datetime
 
-if (
-    not session_with_id.schema.is_activated()
-    and not session_with_datetime.schema.is_activated()
-):
+if session_with_datetime.schema.is_activated():
+    session = session_with_datetime
+elif session_with_id.schema.is_activated():
+    session = session_with_id
+else:
     raise dj.DataJointError("Session schema has not been activated.")
 
-for session_module in [session_with_datetime, session_with_id]:
-    if session_module.schema.is_activated():
-        session = session_module
-    else:
-        continue
-
 
 def session_to_nwb(
     session_key: dict,