Commit 0699d934 authored by narugo1992's avatar narugo1992
Browse files

dev(narugo): add test for new modules

parent 34caffbd

.github/workflows/export.yml

+1 −1
Original line number Diff line number Diff line 
name: Onnx Model Export



on:

  push:

  #  push:

  workflow_dispatch:



jobs:

.github/workflows/test.yml

+7 −0
Original line number Diff line number Diff line
@@ -96,6 +96,13 @@ jobs: 
          tree .

          cloc imgutils

          cloc test

      - name: Prepare dataset

        shell: bash

        env:

          HF_NARUGO_USERNAME: ${{ secrets.HF_NARUGO_USERNAME }}

          HF_NARUGO_PASSWORD: ${{ secrets.HF_NARUGO_PASSWORD }}

        run: |

          make dataset

      - name: Run unittest

        env:

          CI: 'true'

.gitignore

+3 −1
Original line number Diff line number Diff line
@@ -1215,3 +1215,5 @@ fabric.properties 
!/resource/yolort/Makefile

!/test/testfile/**/*

!/zoo/testfile/**/*

/test/testfile/dataset

/zoo/testfile/dataset
 
 No newline at end of file

Makefile

+7 −0
Original line number Diff line number Diff line
@@ -8,6 +8,7 @@ BUILD_DIR := ${PROJ_DIR}/build 
DIST_DIR      := ${PROJ_DIR}/dist

TEST_DIR      := ${PROJ_DIR}/test

TESTFILE_DIR  := ${TEST_DIR}/testfile

DATASET_DIR   := ${TESTFILE_DIR}/dataset

SRC_DIR       := ${PROJ_DIR}/imgutils

TEMPLATES_DIR := ${PROJ_DIR}/templates

RESOURCE_DIR  := ${PROJ_DIR}/resource
@@ -45,3 +46,9 @@ docs: 
	$(MAKE) -C "${DOC_DIR}" build

pdocs:

	$(MAKE) -C "${DOC_DIR}" prod



dataset:

	mkdir -p ${DATASET_DIR}

	if [ ! -d ${DATASET_DIR}/chafen_arknights ]; then \

		git clone https://${HF_NARUGO_USERNAME}:${HF_NARUGO_PASSWORD}@huggingface.co/datasets/deepghs/chafen_arknights.git ${DATASET_DIR}/chafen_arknights; \

	fi
 
 No newline at end of file

imgutils/clustering/__init__.py

+1 −0
Original line number Diff line number Diff line 
from .lpips import extract_feature, lpips_difference, batch_lpips_difference