fix using the correct ml-model inst when configuring categories

hbw/config/categories.py

@@ -9,6 +9,7 @@
 import law
 
 from columnflow.config_util import create_category_combinations
+from columnflow.ml import MLModel
 from hbw.util import call_once_on_config
 
 import order as od
@@ -154,6 +155,9 @@ def add_categories_production(config: od.Config) -> None:
 
 @call_once_on_config()
 def add_categories_ml(config, ml_model_inst):
+    # if not already done, get the ml_model instance
+    if isinstance(ml_model_inst, str):
+        ml_model_inst = MLModel.get_cls(ml_model_inst)(config)
 
     # add ml categories directly to the config
     ml_categories = []

hbw/production/categories.py

@@ -11,7 +11,6 @@
 from columnflow.production.categories import category_ids
 
 from hbw.config.categories import add_categories_production, add_categories_ml
-from hbw.ml.dense_classifier import dense_test
 
 np = maybe_import("numpy")
 ak = maybe_import("awkward")
@@ -76,7 +75,7 @@ def ml_cats_init(self: Producer) -> None:
 
     # add categories to config inst
     add_categories_production(self.config_inst)
-    add_categories_ml(self.config_inst, dense_test)
+    add_categories_ml(self.config_inst, self.ml_model_name)
 
 
 # get all the derived DenseClassifier models and instantiate a corresponding producer