fix port for plants

laminlabs · Nov 12, 2024 · a1d2d6a · a1d2d6a
1 parent e75518c
commit a1d2d6a
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/bionty/base/entities/_gene.py b/bionty/base/entities/_gene.py
@@ -100,8 +100,11 @@ def __init__(self, organism: str, version: str, kingdom: Literal['vertibrates',
         self._organism = (
             Organism(version=version, organism=kingdom).lookup().dict().get(organism)  # type:ignore
         )
+        port = 3306
+        if kingdom == "plants":
+            port = 4157
         self._url = (
-            f"mysql+mysqldb://anonymous:@ensembldb.ensembl.org/{self._organism.core_db}"
+            f"mysql+mysqldb://anonymous:@ensembldb.ensembl.org:{port}/{self._organism.core_db}"
         )
         self._engine = create_engine(url=self._url)