Update training.py (#228)

* Update training.py

add _add_tags method

* fix

modelforge/train/training.py

@@ -1173,6 +1173,35 @@ def _replace_placeholder_in_experimental_name(self, experiment_name: str) -> str
         )
         return experiment_name
 
+    def _add_tags(self, tags: List[str]) -> List[str]:
+        """
+        Add tags to the wandb tags.
+
+        Parameters
+        ----------
+        tags : List[str]
+            List of tags to add to the experiment.
+
+        Returns
+        -------
+        List[str]
+            List of tags for the experiment.
+        """
+
+        # add version
+        import modelforge
+
+        tags.append(str(modelforge.__version__))
+        # add dataset
+        tags.append(self.dataset_parameter.dataset_name)
+        # add potential name
+        tags.append(self.potential_parameter.potential_name)
+        # add information about what is included in the loss
+        str_loss_property = "-".join(self.training_parameter.loss_parameter.loss_property)
+        tags.append(f"loss-{str_loss_property}")
+
+        return tags
+
 
 from typing import List, Optional, Union