Remove artifacts from final metrics (#216)

* Update wandb.py

* Update wandb.py

* Update wandb.py

* Update _optimization_info.py

* remove print

Author: Samoed

autointent/_callbacks/wandb.py

@@ -112,11 +112,14 @@ def log_final_metrics(self, metrics: dict[str, Any]) -> None:
         }
 
         try:
-            self.wandb.init(config=metrics, **wandb_run_init_args)
+            config = metrics["configs"]
+            self.wandb.init(config=config, **wandb_run_init_args)
+            self.wandb.log(metrics)
         except Exception as e:
             if "run config cannot exceed" not in str(e):
                 # https://github.com/deeppavlov/AutoIntent/issues/202
                 raise
+            self._close_current_run()
             logger.warning("W&B run config is too large, skipping logging modules configs")
             logger.warning("'final_metrics' will be logged to W&B with pipeline_metrics only")
             logger.warning("If you want to access modules configs in future, address to the individual modules runs")

autointent/context/optimization_info/_optimization_info.py

@@ -225,7 +225,6 @@ def dump_evaluation_results(self) -> dict[str, Any]:
             "pipeline_metrics": self.pipeline_metrics,
             "metrics": node_wise_metrics,
             "configs": self.trials.model_dump(),
-            "artifacts": self.artifacts.model_dump(),
         }
 
     def dump(self, path: Path) -> None: