update pyre nightly version (#1076)

Summary:

Pyre signal is failing on github because the nightly version that we're pinned to is no longer available in pip

Differential Revision: D75791451

Author: Daniel Ohayon

.pyre_configuration

@@ -18,5 +18,5 @@
     "stubs"
   ],
   "strict": true,
-  "version": "0.0.101732536891"
+  "version": "0.0.101748862749"
 }

scripts/kfpint.py

@@ -209,7 +209,7 @@ def run_pipeline(build: BuildInfo, pipeline_file: str) -> object:
 
 
 def wait_for_pipeline(
-    resp: Any,  # pyre-fixme: KFP doesn't have a response type
+    resp: Any,  # KFP response object
 ) -> None:
     print(f"{resp.run_id} - waiting for completion")
     result = resp.wait_for_run_completion(

torchx/cli/argparse_util.py

@@ -35,7 +35,6 @@ def __init__(
         dest: str,
         option_strings: Sequence[Text],
         required: bool = False,
-        # pyre-ignore[2] declared as Any in superclass Action
         default: Any = None,
         **kwargs: Any,
     ) -> None:
@@ -69,7 +68,7 @@ def __call__(
         self,
         parser: ArgumentParser,
         namespace: Namespace,
-        values: Any,  # pyre-ignore[2] declared as Any in superclass Action
+        values: Any,
         option_string: Optional[str] = None,
     ) -> None:
         if option_string is not None:
@@ -93,7 +92,6 @@ def __init__(
         dest: str,
         option_strings: Sequence[Text],
         required: bool = False,
-        # pyre-ignore[2] declared as Any in superclass Action
         default: Any = None,
         **kwargs: Any,
     ) -> None:

torchx/schedulers/aws_batch_scheduler.py

@@ -421,9 +421,7 @@ class AWSBatchScheduler(
     def __init__(
         self,
         session_name: str,
-        # pyre-fixme[2]: Parameter annotation cannot be `Any`.
         client: Optional[Any] = None,
-        # pyre-fixme[2]: Parameter annotation cannot be `Any`.
         log_client: Optional[Any] = None,
         docker_client: Optional["DockerClient"] = None,
     ) -> None:
@@ -432,18 +430,13 @@ def __init__(
 
         # pyre-fixme[4]: Attribute annotation cannot be `Any`.
         self.__client = client
-        # pyre-fixme[4]: Attribute annotation cannot be `Any`.
-        self.__log_client = log_client
 
-    @property
-    # pyre-fixme[3]: Return annotation cannot be `Any`.
     def _client(self) -> Any:
         if self.__client:
             return self.__client
         return _local_session().client("batch")
 
     @property
-    # pyre-fixme[3]: Return annotation cannot be `Any`.
     def _log_client(self) -> Any:
         if self.__log_client:
             return self.__log_client
@@ -833,9 +826,7 @@ def _stream_events(
 
 def create_scheduler(
     session_name: str,
-    # pyre-fixme[2]: Parameter annotation cannot be `Any`.
     client: Optional[Any] = None,
-    # pyre-fixme[2]: Parameter annotation cannot be `Any`.
     log_client: Optional[Any] = None,
     docker_client: Optional["DockerClient"] = None,
     **kwargs: object,

torchx/schedulers/aws_sagemaker_scheduler.py

@@ -197,20 +197,23 @@ class AWSSageMakerScheduler(
     def __init__(
         self,
         session_name: str,
-        client: Optional[Any] = None,  # pyre-ignore[2]
+        client: Optional[Any] = None,
         docker_client: Optional["DockerClient"] = None,
     ) -> None:
         super().__init__("aws_sagemaker", session_name, docker_client=docker_client)
         # pyre-fixme[4]: Attribute annotation cannot be `Any`.
         self.__client = client
 
     @property
-    # pyre-fixme[3]: Return annotation cannot be `Any`.
     def _client(self) -> Any:
         if self.__client:
             return self.__client
         return _local_session().client("sagemaker")
 
+    @property
+    def _client(self) -> Any:
+        return _local_session().client("sagemaker")
+
     def schedule(self, dryrun_info: AppDryRunInfo[AWSSageMakerJob]) -> str:
         cfg = dryrun_info._cfg
         assert cfg is not None, f"{dryrun_info} missing cfg"

torchx/schedulers/test/ray_scheduler_test.py

@@ -216,7 +216,6 @@ def test_submit_dryrun_raises_error_if_cluster_config_file_is_not_found(
         ):
             self._scheduler._submit_dryrun(self._app_def, self._run_cfg)
 
-    # pyre-ignore[2]: Parameter `value` must have a type other than `Any`
     def _assert_config_value(self, name: str, value: Any, type_name: str) -> None:
         # pyre-fixme: TypedDict indexes by string literal
         self._run_cfg[name] = value

torchx/specs/api.py

@@ -391,7 +391,6 @@ class Role:
     )
     overrides: Dict[str, Any] = field(default_factory=dict)
 
-    # pyre-ignore
     def __getattribute__(self, attrname: str) -> Any:
         if attrname == "overrides":
             return super().__getattribute__(attrname)

torchx/specs/builders.py

@@ -31,7 +31,7 @@ def _create_args_parser(
 
 
 def _create_args_parser_from_parameters(
-    cmpnt_fn: Callable[..., Any],  # pyre-ignore[2]
+    cmpnt_fn: Callable[..., Any],
     parameters: Mapping[str, inspect.Parameter],
     cmpnt_defaults: Optional[Dict[str, str]] = None,
     config: Optional[Dict[str, Any]] = None,
@@ -112,7 +112,7 @@ def _merge_config_values_with_args(
 
 
 def parse_args(
-    cmpnt_fn: Callable[..., Any],  # pyre-ignore[2]
+    cmpnt_fn: Callable[..., Any],
     cmpnt_args: List[str],
     cmpnt_defaults: Optional[Dict[str, Any]] = None,
     config: Optional[Dict[str, Any]] = None,
@@ -141,7 +141,7 @@ def parse_args(
 
 
 def materialize_appdef(
-    cmpnt_fn: Callable[..., Any],  # pyre-ignore[2]
+    cmpnt_fn: Callable[..., Any],
     cmpnt_args: List[str],
     cmpnt_defaults: Optional[Dict[str, Any]] = None,
     config: Optional[Dict[str, Any]] = None,

torchx/specs/finder.py

@@ -55,7 +55,6 @@ class _Component:
     description: str
     fn_name: str
 
-    # pyre-ignore[4] TODO temporary until PipelineDef is decoupled and can be exposed as type to OSS
     fn: Callable[..., Any]
 
     validation_errors: List[str]

torchx/specs/test/builders_test.py

@@ -428,7 +428,6 @@ def test_single_letter(self) -> None:
             "arg2",
         )
 
-    # pyre-ignore[3]
     def _get_argument_help(
         self, parser: argparse.ArgumentParser, name: str
     ) -> Optional[Tuple[str, Any]]: