From 7ff4f298989b11dda530e7a6eafd29fb7a53c5ea Mon Sep 17 00:00:00 2001
From: Oliver Mannion <125105+tekumara@users.noreply.github.com>
Date: Sat, 2 Dec 2023 16:54:33 +1100
Subject: [PATCH] ci: make check

---
 .github/workflows/ci.yml | 2 +-
 flows/ping_ray.py        | 3 +++
 flows/ray_flow.py        | 3 +++
 3 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml
index d5a46cd..3d665e8 100644
--- a/.github/workflows/ci.yml
+++ b/.github/workflows/ci.yml
@@ -26,7 +26,7 @@ jobs:
       - name: Run make install
         if: steps.cache-venv.outputs.cache-hit != 'true'
         run: make install
-      - run: make test
+      - run: make check test
       - name: install k3d
         run: curl -s https://raw.githubusercontent.com/k3d-io/k3d/main/install.sh | bash
       # - uses: mxschmitt/action-tmate@v3
diff --git a/flows/ping_ray.py b/flows/ping_ray.py
index 390ab94..0bc8459 100644
--- a/flows/ping_ray.py
+++ b/flows/ping_ray.py
@@ -1,3 +1,6 @@
+# allow missing imports because we may not have ray installed
+# pyright: reportMissingImports=false
+
 import grpc
 import grpc._channel
 import ray.core.generated.ray_client_pb2 as ray_client_pb2
diff --git a/flows/ray_flow.py b/flows/ray_flow.py
index 0f08569..6dbcd00 100644
--- a/flows/ray_flow.py
+++ b/flows/ray_flow.py
@@ -1,3 +1,6 @@
+# allow missing imports because we may not have ray installed
+# pyright: reportMissingImports=false
+
 import sys
 
 from prefect import flow, get_run_logger, task