remove missed const

choderalab · Sep 30, 2024 · 71ede46 · 71ede46
1 parent 5808453
commit 71ede46
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mtenn/config.py b/mtenn/config.py
@@ -140,7 +140,8 @@ class ModelConfigBase(BaseModel):
     to implement the ``_build`` method in order to be used.
     """
 
-    model_type: ModelType = Field(ModelType.INVALID, const=True, frozen=True)
+    model_type: Literal[ModelType.INVALID] = ModelType.INVALID
+
 
     # Random seed optional for reproducibility
     rand_seed: int | None = Field(