kaurm43 commited on
Commit
60f78c2
·
verified ·
1 Parent(s): ef431a2

Update PolyAgent/orchestrator.py

Browse files
Files changed (1) hide show
  1. PolyAgent/orchestrator.py +1 -1
PolyAgent/orchestrator.py CHANGED
@@ -1636,7 +1636,7 @@ class PolymerOrchestrator:
1636
  def _ensure_cl_encoder(self):
1637
  if self._cl_encoder is None:
1638
  try:
1639
- from PolyFusion.GINE import GineBlock, MaskedGINE, match_edge_attr_to_index, safe_get
1640
  from PolyFusion.SchNet import NodeSchNetWrapper
1641
  from PolyFusion.Transformer import PooledFingerprintEncoder as FingerprintEncoder
1642
  from PolyFusion.DeBERTav2 import PSMILESDebertaEncoder, build_psmiles_tokenizer
 
1636
  def _ensure_cl_encoder(self):
1637
  if self._cl_encoder is None:
1638
  try:
1639
+ from PolyFusion.GINE import GineEncoder, GineBlock, MaskedGINE, match_edge_attr_to_index, safe_get
1640
  from PolyFusion.SchNet import NodeSchNetWrapper
1641
  from PolyFusion.Transformer import PooledFingerprintEncoder as FingerprintEncoder
1642
  from PolyFusion.DeBERTav2 import PSMILESDebertaEncoder, build_psmiles_tokenizer