From 088bd5b462b4e0990b21da687399d0e8b67c8697 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Sat, 17 Sep 2022 00:50:20 +0200
Subject: [PATCH] Create ci.yaml (#10)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
---
 .github/workflows/ci.yaml | 66 +++++++++++++++++++++++++++++++++++++++
 setup.py                  |  5 ++-
 2 files changed, 68 insertions(+), 3 deletions(-)
 create mode 100644 .github/workflows/ci.yaml

diff --git a/.github/workflows/ci.yaml b/.github/workflows/ci.yaml
new file mode 100644
index 0000000..ddaaff1
--- /dev/null
+++ b/.github/workflows/ci.yaml
@@ -0,0 +1,66 @@
+# Ultralytics Continuous Integration (CI) GitHub Actions tests
+
+name: Ultralytics CI
+
+on:
+  push:
+    branches: [ main ]
+  pull_request:
+    branches: [ main ]
+  # schedule:
+  #   - cron: '0 0 * * *'  # runs at 00:00 UTC every day
+
+jobs:
+  Tests:
+    runs-on: ${{ matrix.os }}
+    strategy:
+      fail-fast: false
+      matrix:
+        os: [ ubuntu-latest ]
+        python-version: [ 3.9 ]
+    timeout-minutes: 60
+    steps:
+      - uses: actions/checkout@v3
+      - uses: actions/setup-python@v3
+        with:
+          python-version: ${{ matrix.python-version }}
+      - name: Get cache dir
+        # https://github.com/actions/cache/blob/master/examples.md#multiple-oss-in-a-workflow
+        id: pip-cache
+        run: echo "::set-output name=dir::$(pip cache dir)"
+      - name: Cache pip
+        uses: actions/cache@v3
+        with:
+          path: ${{ steps.pip-cache.outputs.dir }}
+          key: ${{ runner.os }}-${{ matrix.python-version }}-pip-${{ hashFiles('requirements.txt') }}
+          restore-keys: ${{ runner.os }}-${{ matrix.python-version }}-pip-
+      - name: Install requirements
+        run: |
+          python -m pip install --upgrade pip wheel
+          pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cpu
+          # pip install ultralytics (production)
+          # python setup.py install (deprecated)
+          python -m pip install .
+      - name: Check environment
+        run: |
+          echo "RUNNER_OS is ${{ runner.os }}"
+          echo "GITHUB_EVENT_NAME is ${{ github.event_name }}"
+          echo "GITHUB_WORKFLOW is ${{ github.workflow }}"
+          echo "GITHUB_ACTOR is ${{ github.actor }}"
+          echo "GITHUB_REPOSITORY is ${{ github.repository }}"
+          echo "GITHUB_REPOSITORY_OWNER is ${{ github.repository_owner }}"
+          python --version
+          pip --version
+          pip list
+      - name: Test HUB training
+        shell: python
+        env:
+         APIKEY: ${{ secrets.ULTRALYTICS_HUB_APIKEY }}
+        run: |
+          import os
+          import ultralytics
+          key = os.environ['APIKEY']
+          print(ultralytics.__version__)
+          # ultralytics.checks()
+          # ultralytics.reset_model(key)  # reset trained model
+          # ultralytics.start(key)  # train model
diff --git a/setup.py b/setup.py
index 580da68..967656d 100644
--- a/setup.py
+++ b/setup.py
@@ -2,7 +2,7 @@ import re
 from pathlib import Path
 
 import pkg_resources as pkg
-from setuptools import find_packages, setup
+from setuptools import setup
 
 # Settings
 FILE = Path(__file__).resolve()
@@ -29,8 +29,7 @@ setup(
         'Source': 'https://github.com/ultralytics/ultralytics',},
     author="Ultralytics",
     author_email='hello@ultralytics.com',
-    # package_dir={'': 'ultralytics'},  # Optional, use if source code is in a subdirectory under the project root, i.e. `src/`
-    packages=find_packages('ultralytics'),  # required
+    packages=['ultralytics'],  # required
     include_package_data=True,
     install_requires=REQUIREMENTS,
     extras_require={