fix python test

deephaven · Sep 3, 2024 · 2b09ce5 · 2b09ce5
1 parent b58c7db
commit 2b09ce5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/py/server/deephaven/experimental/keyed_record_adapter.py b/py/server/deephaven/experimental/keyed_record_adapter.py
@@ -51,9 +51,9 @@ def __init__(self,
 
         # Build a list of the DTypes corresponding to the key columns
         if self.is_single_key_col:
-            self.key_dtypes = self.table.columns_dict()[key_cols[0]].data_type
+            self.key_dtypes = self.table.definition[key_cols[0]].data_type
         else:
-            self.key_dtypes = [self.table.columns_dict()[col_name].data_type for col_name in key_cols]
+            self.key_dtypes = [self.table.definition[col_name].data_type for col_name in key_cols]
 
     def get_records(self, data_keys: [typing.List[K], numpy.ndarray]) -> [typing.Dict[K, T], typing.List[T]]:
         """