Merge pull request #922 from facebookresearch/small-fixes

Extending some 1.1 changes
facebookresearch · Oct 28, 2022 · 66be164 · 66be164
2 parents a4aa704 + 3ad6a49
commit 66be164
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/.github/workflows/deploy-docs.yml b/.github/workflows/deploy-docs.yml
@@ -21,11 +21,6 @@ jobs:
           node-version: 16.16.0
           cache: yarn
       - uses: actions/setup-python@v2
-      - name: Install Poetry
-        run: |
-
-      - name: Install dependencies
-        run: poetry install --no-interaction
       - name: 🤖 Install Mephisto
         run: |
           cd ../../

diff --git a/mephisto/abstractions/_subcomponents/task_runner.py b/mephisto/abstractions/_subcomponents/task_runner.py
@@ -186,6 +186,9 @@ def _launch_and_run_onboarding(
                 AgentDisconnectedError,
                 AgentShutdownError,
             ):
+                if onboarding_agent.get_status() not in AgentState.complete():
+                    # Absent agents at this stage should be disconnected
+                    onboarding_agent.update_status(AgentState.STATUS_DISCONNECT)
                 self.cleanup_onboarding(onboarding_agent)
             except Exception as e:
                 logger.exception(