- pyyaml instead of yaml

- union instead of pipe

Author: schnamo

chebai/loss/semantic.py

@@ -4,7 +4,7 @@
 
 import math
 import torch
-from typing import Literal
+from typing import Literal, Union
 
 from chebai.preprocessing.datasets.chebi import _ChEBIDataExtractor, ChEBIOver100
 from chebai.preprocessing.datasets.pubchem import LabeledUnlabeledMixed
@@ -14,7 +14,7 @@
 class ImplicationLoss(torch.nn.Module):
     def __init__(
         self,
-        data_extractor: _ChEBIDataExtractor | LabeledUnlabeledMixed,
+        data_extractor: Union[_ChEBIDataExtractor, LabeledUnlabeledMixed],
         base_loss: torch.nn.Module = None,
         tnorm: Literal["product", "lukasiewicz", "xu19"] = "product",
         impl_loss_weight=0.1,  # weight of implication loss in relation to base_loss
@@ -114,7 +114,7 @@ class DisjointLoss(ImplicationLoss):
     def __init__(
         self,
         path_to_disjointness,
-        data_extractor: _ChEBIDataExtractor | LabeledUnlabeledMixed,
+        data_extractor: Union[_ChEBIDataExtractor, LabeledUnlabeledMixed],
         base_loss: torch.nn.Module = None,
         disjoint_loss_weight=100,
         **kwargs,

chebai/result/analyse_sem.py

@@ -11,6 +11,7 @@
 from torchmetrics.functional.classification import multilabel_f1_score
 import wandb
 import gc
+from typing import List,Union
 from utils import *
 
 DEVICE = "cpu"  # torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
@@ -244,7 +245,7 @@ def analyse_run(
     labeled_data_cls=ChEBIOver100,  # use labels from this dataset for violations
     chebi_version=231,
     results_path=os.path.join("_semantic", "eval_results.csv"),
-    violation_metrics: [str | list[callable]] = "all",
+    violation_metrics: Union[str, List[callable]] = "all",
     verbose_violation_output=False,
 ):
     """Calculates all semantic metrics for given predictions (and supervised metrics if labels are provided),

setup.py

@@ -48,7 +48,7 @@
         "iterative-stratification",
         "wandb",
         "chardet",
-        "yaml",
+        "pyyaml",
         "torchmetrics",
     ],
     extras_require={"dev": ["black", "isort", "pre-commit"]},