From 1a8050366fa6789d6bfaaad58419a22ce2ae2c55 Mon Sep 17 00:00:00 2001
From: Ben Viggiano <viggiano@stanford.edu>
Date: Wed, 27 May 2026 17:48:43 -0700
Subject: [PATCH] Type max_inference_sigma as float | None to match downstream
 consumer

---
 esm/models/esmfold2/processor.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/esm/models/esmfold2/processor.py b/esm/models/esmfold2/processor.py
index 0aaf596..3616188 100644
--- a/esm/models/esmfold2/processor.py
+++ b/esm/models/esmfold2/processor.py
@@ -294,7 +294,7 @@ def fold(
         seed: int | None = None,
         noise_scale: float | None = None,
         step_scale: float | None = None,
-        max_inference_sigma: int | None = None,
+        max_inference_sigma: float | None = None,
         early_exit: bool = False,
         complex_id: str = "pred",
     ) -> MolecularComplexResult | list[MolecularComplexResult]: