From 42e7429f023a86e445db2b978e9c86d5286455ba Mon Sep 17 00:00:00 2001 From: Liyu Xiao Date: Wed, 18 Jun 2025 13:52:10 -0400 Subject: [PATCH] remove warnings --- .riahub/workflows/workflow.yaml | 3 +-- scripts/training/plot_data.py | 2 +- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/.riahub/workflows/workflow.yaml b/.riahub/workflows/workflow.yaml index e3abb73..de58f39 100644 --- a/.riahub/workflows/workflow.yaml +++ b/.riahub/workflows/workflow.yaml @@ -120,8 +120,7 @@ jobs: - name: 6. Convert to ORT file run: | - python -m scripts/ort/convert_to_ort.py - + PYTHONPATH=. python scripts/ort/convert_to_ort.py - name: Upload ORT file uses: actions/upload-artifact@v3 diff --git a/scripts/training/plot_data.py b/scripts/training/plot_data.py index 3579970..5b47c1b 100644 --- a/scripts/training/plot_data.py +++ b/scripts/training/plot_data.py @@ -2,7 +2,7 @@ import os import torch import numpy as np from sklearn.metrics import classification_report -import matplotlib +os.environ["NNPACK"] = "0" from matplotlib import pyplot as plt from scripts.training.mobilenetv3 import mobilenetv3, RFClassifier