Skip to content

Commit

Permalink
Apply suggestions from code review
Browse files Browse the repository at this point in the history
Co-authored-by: Justin Chu <justinchuby@users.noreply.github.com>
Signed-off-by: Christian Bourjau <cbourjau@users.noreply.github.com>
  • Loading branch information
cbourjau and justinchuby committed Apr 15, 2024
1 parent b6233e9 commit 6a81b80
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions onnx/backend/test/stat_coverage.py
Expand Up @@ -5,7 +5,7 @@
# SPDX-License-Identifier: Apache-2.0

import os
from pathlib import Path
import pathlib
from typing import IO, Any, Dict, List, Sequence

from onnx import AttributeProto, defs, load
Expand Down Expand Up @@ -154,7 +154,7 @@ def gen_model_test_coverage(
attrs: Dict[str, Dict[str, List[Any]]] = {}
model_paths: List[Any] = []
for rt in load_model_tests(
data_dir=str(Path(__file__).parent / "data"), kind="real"
data_dir=path.Path(__file__).parent / "data", kind="real"
):
if rt.url.startswith("onnx/backend/test/data/light/"):
# testing local files
Expand Down
4 changes: 2 additions & 2 deletions tests/backend/test_backend_onnxruntime.py
Expand Up @@ -6,7 +6,7 @@

import platform
import unittest
from pathlib import Path
import pathlib
from typing import Any

import numpy
Expand All @@ -27,7 +27,7 @@
ort: Any = None # type: ignore[no-redef]
ort_version: Any = None # type: ignore[no-redef]

TEST_DATA_DIR = Path(__file__).parent.parent.parent / "onnx/backend/test/data"
TEST_DATA_DIR = pathlib.Path(__file__).parent.parent.parent / "onnx/backend/test/data"

# The following just executes a backend based on InferenceSession through the backend test

Expand Down

0 comments on commit 6a81b80

Please sign in to comment.