Skip to content

Commit

Permalink
Pyre Configurationless migration for] [batch:231/244]
Browse files Browse the repository at this point in the history
Differential Revision: D54562955

Pull Request resolved: #845
  • Loading branch information
connernilsen committed Mar 8, 2024
1 parent dac181d commit 380fb8d
Show file tree
Hide file tree
Showing 157 changed files with 314 additions and 0 deletions.
2 changes: 2 additions & 0 deletions scripts/collect_env.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

# This script uses https://raw.githubusercontent.com/pytorch/pytorch/master/torch/utils/collect_env.py
# and collects additional information on top of it to output relevant system
# environment info.
Expand Down
2 changes: 2 additions & 0 deletions scripts/component_integration_tests.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

"""
Kubernetes integration tests.
"""
Expand Down
2 changes: 2 additions & 0 deletions scripts/example_app_defs.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

"""
App Defs for integration tests.
"""
Expand Down
2 changes: 2 additions & 0 deletions scripts/integ_test_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import dataclasses
import os
import subprocess
Expand Down
2 changes: 2 additions & 0 deletions scripts/kfpint.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

"""
This file runs the KFP integration tests on KFP cluster. There's a number of
environment variables that need to be setup as well as the cluster.
Expand Down
2 changes: 2 additions & 0 deletions scripts/kube_dist_trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

"""
Kubernetes integration tests.
"""
Expand Down
2 changes: 2 additions & 0 deletions torchx/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

from .version import ( # noqa F401; noqa F401
__version__ as __version__,
TORCHX_IMAGE as IMAGE,
Expand Down
2 changes: 2 additions & 0 deletions torchx/apps/serve/serve.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import binascii
import os
Expand Down
2 changes: 2 additions & 0 deletions torchx/apps/serve/test/serve_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import os.path
import tempfile
import unittest
Expand Down
2 changes: 2 additions & 0 deletions torchx/apps/utils/booth_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import sys
from typing import List
Expand Down
2 changes: 2 additions & 0 deletions torchx/apps/utils/copy_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import os
import shutil
Expand Down
2 changes: 2 additions & 0 deletions torchx/apps/utils/process_monitor.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import subprocess
import sys
Expand Down
2 changes: 2 additions & 0 deletions torchx/apps/utils/test/booth_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import shutil
import tempfile
import unittest
Expand Down
2 changes: 2 additions & 0 deletions torchx/apps/utils/test/copy_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import os
import os.path
import tempfile
Expand Down
2 changes: 2 additions & 0 deletions torchx/apps/utils/test/process_monitor_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import io
import unittest
from unittest.mock import MagicMock, patch
Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

"""
The ``torchx`` CLI is a commandline tool around :py:class:`torchx.runner.Runner`.
It allows users to launch :py:class:`torchx.specs.AppDef` directly onto
Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/argparse_util.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

from argparse import Action, ArgumentParser, Namespace
from typing import Any, Dict, Optional, Sequence, Text

Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/cmd_base.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import abc
import argparse

Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/cmd_cancel.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import logging

Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/cmd_configure.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import logging
import sys
Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/cmd_describe.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import dataclasses
import logging
Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/cmd_list.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import logging

Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/cmd_log.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import logging
import re
Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/cmd_run.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import logging
import os
Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/cmd_runopts.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import logging

Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/cmd_status.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import logging
import sys
Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/cmd_tracker.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import logging

Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/colors.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import sys

# only print colors if outputting directly to a terminal
Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import logging
import os
import sys
Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/test/argparse_util_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

from argparse import ArgumentParser
from unittest import mock

Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/test/cmd_cancel_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import unittest
from unittest.mock import MagicMock, patch
Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/test/cmd_configure_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import os
import shutil
Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/test/cmd_describe_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import unittest
from unittest.mock import patch
Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/test/cmd_list_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import unittest
from unittest.mock import MagicMock, patch
Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/test/cmd_log_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import io
import sys
import unittest
Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/test/cmd_run_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import dataclasses
import io
Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/test/cmd_runopts_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import unittest

Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/test/cmd_status_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import unittest
from unittest.mock import patch
Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/test/cmd_tracker_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import unittest
from unittest.mock import MagicMock, patch
Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/test/components.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict


import torchx.specs as specs

Expand Down
2 changes: 2 additions & 0 deletions torchx/cli/test/container/ps_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,4 +5,6 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

print("ps_main")
2 changes: 2 additions & 0 deletions torchx/cli/test/container/reader_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,4 +5,6 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

print("reader_main")
2 changes: 2 additions & 0 deletions torchx/cli/test/container/train_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,4 +5,6 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

print("train_main")
2 changes: 2 additions & 0 deletions torchx/cli/test/main_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,6 +5,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

import argparse
import os
import unittest
Expand Down
2 changes: 2 additions & 0 deletions torchx/components/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,8 @@
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

# pyre-strict

"""
This module contains a collection of builtin TorchX components. The directory
structure is organized by component category. Components are simply
Expand Down
Loading

0 comments on commit 380fb8d

Please sign in to comment.