Simplify inplace column assignment

emdgroup · Dec 16, 2024 · 3bedbc3 · 3bedbc3
1 parent e9b5db0
commit 3bedbc3
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/baybe/simulation/lookup.py b/baybe/simulation/lookup.py
@@ -82,8 +82,7 @@ def _look_up_targets_from_callable(
 ) -> None:
     """Look up target values by querying a callable."""
     df_targets = lookup(queries)
-    for col in df_targets:
-        queries[col] = df_targets[col]
+    queries[df_targets.columns] = df_targets
 
 
 def _look_up_targets_from_dataframe(