Update PolyAgent/orchestrator.py

PolyAgent/orchestrator.py

@@ -1636,10 +1636,10 @@ class PolymerOrchestrator:
     def _ensure_cl_encoder(self):
         if self._cl_encoder is None:
             try:
-                from PolyFusion.GINE import GineBlock, MaskedGINE
-                from PolyFusion.SchNet import NodeSchNet, MaskedSchNet
-                from PolyFusion.Transformer import FingerprintEncoder, MaskedFingerprintModel
-                from PolyFusion.DeBERTav2 import build_model_and_trainer, build_tokenizer
+                from PolyFusion.GINE import GineBlock, MaskedGINE, match_edge_attr_to_index, safe_get
+                from PolyFusion.SchNet import NodeSchNetWrapper
+                from PolyFusion.Transformer import PooledFingerprintEncoder as FingerprintEncoder
+                from PolyFusion.DeBERTav2 import PSMILESDebertaEncoder, build_psmiles_tokenizer
                 from PolyFusion.CL import MultimodalContrastiveModel
             except Exception:
                 raise RuntimeError("Modules not available in python path")