Add predict.py int-test

replicate · Jan 17, 2025 · 95f8404 · 95f8404
1 parent a2611e9
commit 95f8404
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/test-integration/test_integration/fixtures/pydantic2/predict.py b/test-integration/test_integration/fixtures/pydantic2/predict.py
@@ -0,0 +1,6 @@
+from cog import BasePredictor
+
+
+class Predictor(BasePredictor):
+    def predict(self, s: str) -> str:
+        return "hello " + s