Add futures annotations int test (#2405)

* Add futures annotations int test

* Add integration test for futures annotation

---------

Signed-off-by: Will Sackfield <[email protected]>

Author: 8W9aG

test-integration/test_integration/fixtures/future-annotations-project/cog.yaml

@@ -1,3 +1,6 @@
 build:
   python_version: "3.11"
+  python_packages:
+    - pillow==10.0.0
+
 predict: "predict.py:Predictor"

test-integration/test_integration/fixtures/future-annotations-project/predict.py

@@ -0,0 +1,17 @@
+# predict.py - This will FAIL with the TypeError
+from __future__ import annotations
+
+from cog import BasePredictor, Input, Path
+from PIL import Image
+
+class Predictor(BasePredictor):
+    def predict(
+        self,
+        image: Path = Input(description="Input image"),
+    ) -> Path:
+        """Simple pass-through prediction that exhibits the bug."""
+        # This would normally just return the input image
+        img = Image.open(image)
+        output_path = Path("/tmp/output.png")
+        img.save(output_path)
+        return output_path

test-integration/test_integration/fixtures/future-annotations-project/some_image.jpg

@@ -787,3 +787,17 @@ def test_predict_glb_file(cog_binary):
         timeout=120.0,
     )
     assert result.returncode == 0
+
+ 
+def test_predict_future_annotations(cog_binary):
+    project_dir = Path(__file__).parent / "fixtures/future-annotations-project"
+
+    result = subprocess.run(
+        [cog_binary, "predict", "--debug", "-i", "image=@some_image.jpg"],
+        cwd=project_dir,
+        check=True,
+        capture_output=True,
+        text=True,
+        timeout=120.0,
+    )
+    assert result.returncode == 0