From 8967271357b96898da4bd54f3f443fe91f043995 Mon Sep 17 00:00:00 2001
From: Hansong Zhang <hsz@meta.com>
Date: Mon, 23 Oct 2023 16:05:41 -0700
Subject: [PATCH] Simplify gather_models step (#1069)

Summary:
We don't need to install the full ExecuTorch package; we don't need to checkout submodule or use pip cache.

Pull Request resolved: https://github.com/pytorch/executorch/pull/1069

Reviewed By: huydhn

Differential Revision: D50570501

Pulled By: kirklandsign

fbshipit-source-id: 919f94517e508bcf6f5ae4f7b198b967d4eff371
---
 .github/workflows/pull.yml  | 9 +--------
 examples/models/__init__.py | 3 ---
 2 files changed, 1 insertion(+), 11 deletions(-)

diff --git a/.github/workflows/pull.yml b/.github/workflows/pull.yml
index dcd0ecfa82..83e15ffd33 100644
--- a/.github/workflows/pull.yml
+++ b/.github/workflows/pull.yml
@@ -20,22 +20,15 @@ jobs:
     steps:
       - uses: actions/checkout@v3
         with:
-          submodules: 'true'
+          submodules: 'false'
       - uses: actions/setup-python@v4
         with:
           python-version: '3.10'
-          cache: pip
       - name: Extract the list of models to test
         id: gather-models
         run: |
           set -eux
 
-          source .ci/scripts/utils.sh
-          # This is a simple Python script but as it tries to import executorch.examples.models,
-          # it requires a whole bunch of ExecuTorch dependencies on the Docker image
-          install_pip_dependencies
-          install_executorch
-
           PYTHONPATH="${PWD}" python .ci/scripts/gather_test_models.py --event "${GITHUB_EVENT_NAME}"
 
   test-models-linux:
diff --git a/examples/models/__init__.py b/examples/models/__init__.py
index 0f806d34fa..d82b015365 100644
--- a/examples/models/__init__.py
+++ b/examples/models/__init__.py
@@ -4,8 +4,6 @@
 # This source code is licensed under the BSD-style license found in the
 # LICENSE file in the root directory of this source tree.
 
-from .model_factory import EagerModelFactory
-
 MODEL_NAME_TO_MODEL = {
     "mul": ("toy_model", "MulModule"),
     "linear": ("toy_model", "LinearModule"),
@@ -30,6 +28,5 @@
 }
 
 __all__ = [
-    "EagerModelFactory",
     "MODEL_NAME_TO_MODEL",
 ]