Internal change.

PiperOrigin-RevId: 836759014

Author: iindyk

grain/_src/core/BUILD

@@ -85,12 +85,6 @@ py_test(
     ],
 )
 
-py_library(
-    name = "usage_logging",
-    srcs = ["usage_logging.py"],
-    srcs_version = "PY3",
-)
-
 py_library(
     name = "transforms",
     srcs = ["transforms.py"],

grain/_src/core/usage_logging.py

@@ -12,7 +12,6 @@ py_library(
         "//grain/_src/core:monitoring",
         "@pypi//etils:pkg",
         "@abseil-py//absl/logging",
-        "//grain/_src/core:usage_logging",
         "//grain/_src/python/dataset:stats",
     ] + select({
         "@platforms//os:windows": [],
@@ -166,7 +165,6 @@ py_library(
         "//grain/_src/core:sharding",
         "//grain/_src/core:transforms",
         "//grain/_src/core:tree_lib",
-        "//grain/_src/core:usage_logging",
         "//grain/_src/python/dataset",
         "@abseil-py//absl/logging",
         "@pypi//etils:pkg",
@@ -277,7 +275,6 @@ py_library(
         "//grain/_src/core:monitoring",
         "//grain/_src/core:sharding",
         "//grain/_src/core:transforms",
-        "//grain/_src/core:usage_logging",
     ],
 )
 

grain/_src/python/BUILD

@@ -30,7 +30,6 @@
 from grain._src.core import sharding
 from grain._src.core import transforms
 from grain._src.core import tree_lib
-from grain._src.core import usage_logging
 import multiprocessing as mp
 from grain._src.python import checkpointing
 from grain._src.python import operations as ops
@@ -369,7 +368,6 @@ def __init__(
       enable_profiling: If True, profiling info is logged. Note, it only
         supports worker_count >= 1 at the moment.
     """
-    usage_logging.log_event("PyGrainDataLoader", tag_3="PyGrain")
     _api_usage_counter.Increment("DataLoader")
     if worker_count and worker_count < 0:
       raise ValueError(

grain/_src/python/data_loader.py

@@ -36,7 +36,6 @@
 from etils import epath
 
 from grain._src.core import monitoring as grain_monitoring
-from grain._src.core import usage_logging
 from grain._src.python.dataset import stats as dataset_stats
 
 from grain._src.core import monitoring  # pylint: disable=g-bad-import-order

grain/_src/python/data_sources.py

@@ -51,7 +51,6 @@ py_library(
         "//grain/_src/core:sharding",
         "//grain/_src/core:transforms",
         "//grain/_src/core:tree_lib",
-        "//grain/_src/core:usage_logging",
         "//grain/_src/python:checkpointing",
         "//grain/_src/python:grain_pool",
         "//grain/_src/python:options",

grain/_src/python/dataset/BUILD

@@ -56,7 +56,6 @@
 from etils import epath
 from grain._src.core import monitoring as grain_monitoring
 from grain._src.core import transforms
-from grain._src.core import usage_logging
 from grain._src.python import checkpointing
 from grain._src.python import options as grain_options
 from grain._src.python.dataset import base
@@ -358,7 +357,6 @@ def __init__(self, parents: MapDataset | Sequence[MapDataset] = ()):
       parents = tuple(parents)
     super().__init__(parents)
     self._parents = cast(Sequence[MapDataset], self._parents)
-    usage_logging.log_event("MapDataset", tag_3="PyGrain")
     _api_usage_counter.Increment("MapDataset")
 
   @property
@@ -977,7 +975,6 @@ def __init__(
     self._parents = cast(
         Sequence[Union[MapDataset, IterDataset]], self._parents
     )
-    usage_logging.log_event("IterDataset", tag_3="PyGrain")
     _api_usage_counter.Increment("IterDataset")
 
   @property

grain/_src/python/dataset/dataset.py

@@ -5,7 +5,6 @@
 from grain._src.core import monitoring as grain_monitoring
 from grain._src.core import sharding
 from grain._src.core import transforms
-from grain._src.core import usage_logging
 from grain._src.python import data_loader
 from grain._src.python import data_sources
 from grain._src.python import options
@@ -55,7 +54,6 @@ def load(
   Returns:
     DataLoader for this dataset.
   """
-  usage_logging.log_event("load", tag_3="PyGrain")
   _api_usage_counter.Increment("load")
   sampler = samplers.IndexSampler(
       num_records=len(source),