Remove int conversion for new int-based comp_df of task parameter

Hrovatin · Hrovatin · commit bee32aa60182 · 2025-09-12T13:21:18.000+02:00
diff --git a/baybe/surrogates/gaussian_process/core.py b/baybe/surrogates/gaussian_process/core.py
@@ -218,7 +218,7 @@ def _fit(self, train_x: Tensor, train_y: Tensor) -> None:
                     task_param.comp_df.squeeze(axis=1).at[task_param.active_values[0]]
                 ],
                 "rank": context.n_tasks,
-                "all_tasks": task_param.comp_df.squeeze(axis=1).astype(int).to_list(),
+                "all_tasks": task_param.comp_df.squeeze(axis=1).to_list(),
             }
 
         # construct and fit the Gaussian process

Original file line number	Diff line number	Diff line change
`@@ -218,7 +218,7 @@ def _fit(self, train_x: Tensor, train_y: Tensor) -> None:`
`218`	`218`	`task_param.comp_df.squeeze(axis=1).at[task_param.active_values[0]]`
`219`	`219`	`],`
`220`	`220`	`"rank": context.n_tasks,`
`221`		`- "all_tasks": task_param.comp_df.squeeze(axis=1).astype(int).to_list(),`
	`221`	`+ "all_tasks": task_param.comp_df.squeeze(axis=1).to_list(),`
`222`	`222`	`}`
`223`	`223`
`224`	`224`	`# construct and fit the Gaussian process`