Replace incomplete module markers (#14030)

This commit is contained in:
Sebastian Rittau
2025-05-13 08:53:39 +02:00
committed by GitHub
parent 8b877a6993
commit e08e349699
84 changed files with 89 additions and 199 deletions
+1 -1
View File
@@ -432,4 +432,4 @@ def gather_nd(
name: str | None = None,
bad_indices_policy: Literal["", "DEFAULT", "ERROR", "IGNORE"] = "",
) -> Tensor: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -1,4 +1,3 @@
from _typeshed import Incomplete
from typing import NamedTuple
from tensorflow.config import experimental as experimental
@@ -10,4 +9,4 @@ class PhysicalDevice(NamedTuple):
def list_physical_devices(device_type: None | str = None) -> list[PhysicalDevice]: ...
def get_visible_devices(device_type: None | str = None) -> list[PhysicalDevice]: ...
def set_visible_devices(devices: list[PhysicalDevice] | PhysicalDevice, device_type: None | str = None) -> None: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -1,5 +1,4 @@
import typing_extensions
from _typeshed import Incomplete
from typing import TypedDict
from tensorflow.config import PhysicalDevice
@@ -14,4 +13,4 @@ def reset_memory_stats(device: str) -> None: ...
def get_memory_usage(device: PhysicalDevice) -> int: ...
def get_memory_growth(device: PhysicalDevice) -> bool: ...
def set_memory_growth(device: PhysicalDevice, enable: bool) -> None: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -269,4 +269,4 @@ class TFRecordDataset(Dataset[tf.Tensor]):
@property
def element_spec(self) -> tf.TensorSpec: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -1,4 +1,3 @@
from _typeshed import Incomplete
from collections.abc import Callable, Sequence
from typing import Final, TypeVar
@@ -30,4 +29,4 @@ def sample_from_datasets(
seed: int | None = None,
stop_on_empty_dataset: bool = False,
) -> Dataset[_T1]: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -1,5 +1,3 @@
from _typeshed import Incomplete
from tensorflow.python.distribute.distribute_lib import Strategy as Strategy
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -1,5 +1,3 @@
from _typeshed import Incomplete
from .experimental.coordinator import RemoteValue as RemoteValue
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -1,4 +1,3 @@
from _typeshed import Incomplete
from typing import Generic, TypeVar
from tensorflow._aliases import AnyArray
@@ -9,4 +8,4 @@ class RemoteValue(Generic[_Value_co]):
def fetch(self) -> AnyArray: ...
def get(self) -> _Value_co: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
+1 -1
View File
@@ -54,4 +54,4 @@ quint16: DType
string: DType
def as_dtype(type_value: DTypeLike) -> DType: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -7,4 +7,4 @@ _T_co = TypeVar("_T_co", covariant=True)
class Optional(ABC, Generic[_T_co]):
def __getattr__(self, name: str) -> Incomplete: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -16,4 +16,4 @@ class Mesh:
use_xla_spmd: bool = False,
) -> None: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
+1 -2
View File
@@ -1,4 +1,3 @@
from _typeshed import Incomplete
from collections.abc import Iterable, Mapping
from types import TracebackType
from typing import Literal, NamedTuple
@@ -105,4 +104,4 @@ class RaggedFeature(NamedTuple):
def parse_example(
serialized: TensorCompatible, features: _FeatureSpecs, example_names: Iterable[str] | None = None, name: str | None = None
) -> dict[str, TensorLike]: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
+2 -2
View File
@@ -1,4 +1,4 @@
from _typeshed import Incomplete, StrOrBytesPath
from _typeshed import StrOrBytesPath
from collections.abc import Iterable
def rmtree(path: StrOrBytesPath) -> None: ...
@@ -8,4 +8,4 @@ def exists(path: StrOrBytesPath) -> bool: ...
def copy(src: StrOrBytesPath, dst: StrOrBytesPath, overwrite: bool = False) -> None: ...
def makedirs(path: StrOrBytesPath) -> None: ...
def glob(pattern: str | bytes | Iterable[str | bytes]) -> list[str]: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -1,5 +1,3 @@
from _typeshed import Incomplete
from tensorflow.keras import (
activations as activations,
callbacks as callbacks,
@@ -14,4 +12,4 @@ from tensorflow.keras import (
)
from tensorflow.keras.models import Model as Model
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -1,4 +1,3 @@
from _typeshed import Incomplete
from collections.abc import Callable
from typing import Any, overload
@@ -14,4 +13,4 @@ def get(identifier: None) -> None: ...
def get(identifier: str | dict[str, Any] | Constraint) -> Constraint: ...
@overload
def get(identifier: Callable[[Tensor], Tensor]) -> Callable[[Tensor], Tensor]: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -1,4 +1,3 @@
from _typeshed import Incomplete
from collections.abc import Callable
from typing import Any, overload
from typing_extensions import Self, TypeAlias
@@ -48,4 +47,4 @@ def get(identifier: None) -> None: ...
def get(identifier: str | Initializer | dict[str, Any] | type[Initializer]) -> Initializer: ...
@overload
def get(identifier: Callable[[ShapeLike], Tensor]) -> Callable[[ShapeLike], Tensor]: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -444,4 +444,4 @@ class GaussianDropout(Layer[tf.Tensor, tf.Tensor]):
name: str | None = None,
) -> None: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
+1 -1
View File
@@ -206,4 +206,4 @@ def get(identifier: _FuncT) -> _FuncT: ...
# This is complete with respect to methods documented defined here,
# but many methods get re-exported here from tf.keras.metrics that aren't
# covered yet.
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -1,4 +1,3 @@
from _typeshed import Incomplete
from abc import ABCMeta, abstractmethod
from collections.abc import Callable, Iterable, Sequence
from typing import Any, Literal
@@ -116,4 +115,4 @@ def binary_crossentropy(
def categorical_crossentropy(
y_true: TensorCompatible, y_pred: TensorCompatible, from_logits: bool = False, label_smoothing: float = 0.0, axis: int = -1
) -> Tensor: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
+1 -1
View File
@@ -166,4 +166,4 @@ class Model(Layer[_InputT_contra, _OutputT_co]):
def compile_from_config(self, config: dict[str, Any]) -> Self: ...
def export(self, filepath: str | Path, format: str = "tf_saved_model", verbose: bool = True) -> None: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -4,4 +4,4 @@ from tensorflow.keras.optimizers import legacy as legacy, schedules as schedules
Optimizer = Incomplete
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -1,4 +1,3 @@
from _typeshed import Incomplete
from collections.abc import Callable, Iterable
from typing import Any
from typing_extensions import TypeAlias
@@ -59,4 +58,4 @@ class SGD(Optimizer):
self, learning_rate: _LearningRate = 0.01, momentum: float = 0.0, nesterov: bool = False, name: str = "SGD", **kwargs: Any
) -> None: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -1,4 +1,3 @@
from _typeshed import Incomplete
from collections.abc import Callable
from typing import Any, overload
from typing_extensions import Self, TypeAlias
@@ -19,4 +18,4 @@ def get(identifier: None) -> None: ...
def get(identifier: str | dict[str, Any] | Regularizer) -> Regularizer: ...
@overload
def get(identifier: Callable[[Tensor], Tensor]) -> Callable[[Tensor], Tensor]: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
+1 -2
View File
@@ -1,4 +1,3 @@
from _typeshed import Incomplete
from builtins import bool as _bool
from collections.abc import Iterable
from typing import Literal, overload
@@ -53,4 +52,4 @@ def eye(
name: str | None = None,
) -> Tensor: ...
def band_part(input: TensorCompatible, num_lower: Integer, num_upper: Integer, name: str | None = None) -> Tensor: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
+1 -2
View File
@@ -1,4 +1,3 @@
from _typeshed import Incomplete
from collections.abc import Iterable
from typing import TypeVar, overload
from typing_extensions import TypeAlias
@@ -296,4 +295,4 @@ def count_nonzero(
dtype: DTypeLike = ...,
name: str | None = None,
) -> Tensor: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
+1 -2
View File
@@ -1,4 +1,3 @@
from _typeshed import Incomplete
from collections.abc import Sequence
from typing import Any, Literal, overload
@@ -192,4 +191,4 @@ def safe_embedding_lookup_sparse(
name: str | None = None,
allow_fast_lookup: bool = False,
) -> Tensor: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -1,3 +1 @@
from _typeshed import Incomplete
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -2,4 +2,4 @@ from _typeshed import Incomplete
Strategy = Incomplete
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -1,3 +1 @@
from _typeshed import Incomplete
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -1,5 +1,3 @@
from _typeshed import Incomplete
from tensorflow.python.trackable.base import Trackable
class _ResourceMetaclass(type): ...
@@ -8,4 +6,4 @@ class _ResourceMetaclass(type): ...
# it is needed for the public signatures of some APIs.
class CapturableResource(Trackable, metaclass=_ResourceMetaclass): ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
+1 -2
View File
@@ -1,4 +1,3 @@
from _typeshed import Incomplete
from typing import Literal
from tensorflow import Operation, Tensor
@@ -42,4 +41,4 @@ def ResourceApplyAdam(
use_nesterov: bool = False,
name: str | None = None,
) -> Operation: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
+1 -1
View File
@@ -28,4 +28,4 @@ class SparseTensor(metaclass=ABCMeta):
def __mul__(self, y: _SparseTensorCompatible) -> SparseTensor: ...
def __getattr__(self, name: str) -> Incomplete: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -72,4 +72,4 @@ class CheckpointManager:
def latest_checkpoint(checkpoint_dir: str, latest_filename: str | None = None) -> str: ...
def load_variable(ckpt_dir_or_file: str, name: str) -> np.ndarray[Any, Any]: ...
def list_variables(ckpt_dir_or_file: str) -> list[tuple[str, list[int]]]: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -1,5 +1,4 @@
import abc
from _typeshed import Incomplete
from typing_extensions import Self
from tensorflow.python.trackable.base import Trackable
@@ -10,4 +9,4 @@ class PythonState(Trackable, metaclass=abc.ABCMeta):
@abc.abstractmethod
def deserialize(self, string_value: str) -> Self: ...
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module
@@ -1,5 +1,4 @@
import abc
from _typeshed import Incomplete
from typing import Any, Generic, TypeVar, overload
from typing_extensions import ParamSpec
@@ -28,4 +27,4 @@ class PolymorphicFunction(Callable[_P, _R_co], metaclass=abc.ABCMeta):
GenericFunction = PolymorphicFunction
def __getattr__(name: str) -> Incomplete: ...
def __getattr__(name: str): ... # incomplete module