Fix inconsistent 'get' in torsiondrive test

MolSSI · Nov 14, 2023 · f843437 · f843437
1 parent 031a00e
commit f843437
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/qcfractal/qcfractal/components/torsiondrive/testing_helpers.py b/qcfractal/qcfractal/components/torsiondrive/testing_helpers.py
@@ -154,7 +154,7 @@ def run_test_data(
     assert finished
 
     with storage_socket.session_scope() as session:
-        record = storage_socket.records.get([record_id])[0]
-        assert record["status"] == end_status
+        record = session.get(TorsiondriveRecordORM, record_id)
+        assert record.status == end_status
 
     return record_id