bug fix

Author: voorhs

autointent/_pipeline/_pipeline.py

@@ -84,9 +84,9 @@ def set_config(
         elif isinstance(config, DataConfig):
             self.data_config = config
         elif isinstance(config, HFModelConfig):
-            self.hpo_config = config
-        elif isinstance(config, HPOConfig):
             self.transformer_config = config
+        elif isinstance(config, HPOConfig):
+            self.hpo_config = config
         else:
             assert_never(config)
 

autointent/context/_context.py

@@ -2,9 +2,9 @@
 
 import logging
 from pathlib import Path
-from typing import assert_never
 
 import yaml
+from typing_extensions import assert_never
 
 from autointent import Dataset
 from autointent._callbacks import CallbackHandler, get_callbacks