pyrefly support

Author: yangdanny97

.github/workflows/test.yml

@@ -38,6 +38,9 @@ jobs:
       - name: Run ty on 'pandas-stubs' (using the local stubs) and on the local stubs
         run: poetry run poe ty
 
+      - name: Run pyrefly on the local stubs
+        run: poetry run poe pyrefly
+
       - name: Run pyright on 'tests' (using the local stubs) and on the local stubs
         run: poetry run poe pyright
 

pyproject.toml

@@ -34,6 +34,7 @@ python = ">=3.10"
 types-pytz = ">= 2022.1.1"
 numpy = ">= 1.23.5"
 ty = "^0.0.1a8"
+pyrefly = "^0.21.0"
 
 [tool.poetry.group.dev.dependencies]
 mypy = "1.16.0"
@@ -108,6 +109,10 @@ script = "scripts.test:test(dist=True, type_checker='mypy')"
 help = "Run ty on pandas-stubs"
 script = "scripts.test.run:ty"
 
+[tool.poe.tasks.pyrefly]
+help = "Run pyrefly on pandas-stubs"
+script = "scripts.test.run:pyrefly"
+
 [tool.poe.tasks.pyright]
 help = "Run pyright on 'tests' (using the local stubs) and on the local stubs"
 script = "scripts.test.run:pyright_src"

scripts/test/_step.py

@@ -9,6 +9,10 @@
     name="Run ty on 'pandas-stubs' (using the local stubs) and on the local stubs",
     run=run.ty,
 )
+pyrefly_src = Step(
+    name="Run pyrefly on the local stubs",
+    run=run.pyrefly,
+)
 pyright_src = Step(
     name="Run pyright on 'tests' (using the local stubs) and on the local stubs",
     run=run.pyright_src,

scripts/test/run.py

@@ -156,3 +156,8 @@ def released_mypy():
 def ty():
     cmd = ["ty", "check", "pandas-stubs"]
     subprocess.run(cmd, check=True)
+
+
+def pyrefly():
+    cmd = ["pyrefly", "check", "pandas-stubs"]
+    subprocess.run(cmd, check=True)