Skip to content

Commit

Permalink
Remove usage stats telemetry
Browse files Browse the repository at this point in the history
  • Loading branch information
jamie256 committed Jan 9, 2025
1 parent 89188d7 commit 5f45d5f
Show file tree
Hide file tree
Showing 15 changed files with 2 additions and 261 deletions.
3 changes: 0 additions & 3 deletions python/whylogs/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -26,7 +26,6 @@
reader,
write,
)
from .api.usage_stats import emit_usage as __emit_usage_stats
from .api.whylabs import init
from .core import DatasetProfileView
from .migration.converters import v0_to_v1_view
Expand Down Expand Up @@ -65,5 +64,3 @@ def package_version(package: str = __package__) -> str:
__version__,
init,
]

__emit_usage_stats("import")
8 changes: 2 additions & 6 deletions python/whylogs/api/fugue/__init__.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,3 @@
# flake8: noqa
from whylogs.api.usage_stats import emit_usage
from .profiler import fugue_profile

# This import has a side effect
from .profiler import fugue_profile # type: ignore

emit_usage("fugue")
assert fugue_profile is not None
2 changes: 0 additions & 2 deletions python/whylogs/api/logger/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,6 @@
_log_segment,
)
from whylogs.api.logger.transient import TransientLogger
from whylogs.api.usage_stats import emit_usage
from whylogs.api.whylabs.session.notebook_logger import (
notebook_session_log,
notebook_session_log_comparison,
Expand Down Expand Up @@ -55,7 +54,6 @@ def log(
) -> ResultSet:
if multiple is not None:
result_sets: Dict[str, ResultSet] = {}
emit_usage("multiple")
for alias, data in multiple.items():
result_set = TransientLogger(schema=schema).log(data, trace_id=trace_id)
if dataset_timestamp is not None:
Expand Down
2 changes: 0 additions & 2 deletions python/whylogs/api/pyspark/experimental/profiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,15 +4,13 @@
from typing import Dict, Iterable, Optional, Tuple

import whylogs as why
from whylogs.api.usage_stats import emit_usage
from whylogs.core import DatasetSchema
from whylogs.core.metrics.metrics import conf
from whylogs.core.stubs import pd
from whylogs.core.view.column_profile_view import ColumnProfileView
from whylogs.core.view.dataset_profile_view import DatasetProfileView

logger = getLogger(__name__)
emit_usage("pyspark")

try: # type: ignore
from pyspark.ml.functions import vector_to_array
Expand Down
2 changes: 0 additions & 2 deletions python/whylogs/api/pyspark/experimental/segmented_profiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@
import whylogs as why
from whylogs.api.logger.result_set import ResultSet, SegmentedResultSet
from whylogs.api.pyspark.experimental.profiler import COL_NAME_FIELD, COL_PROFILE_FIELD
from whylogs.api.usage_stats import emit_usage
from whylogs.core import DatasetSchema
from whylogs.core.segment import Segment
from whylogs.core.segmentation_partition import SegmentationPartition
Expand All @@ -17,7 +16,6 @@
from whylogs.core.view.dataset_profile_view import DatasetProfileView

logger = getLogger(__name__)
emit_usage("pyspark")

try: # type: ignore
from pyspark.ml.functions import vector_to_array
Expand Down
218 changes: 0 additions & 218 deletions python/whylogs/api/usage_stats/__init__.py

This file was deleted.

5 changes: 0 additions & 5 deletions python/whylogs/api/whylabs/session/session.py
Original file line number Diff line number Diff line change
Expand Up @@ -77,14 +77,12 @@ def __init__(self, config: SessionConfig) -> None:
If neither exist then this will attempt to create a new session and store the id in the config,
which does require a successful service call to whylabs.
"""
from whylogs.api.usage_stats import emit_usage

super().__init__(config)

# Using lazy initialization to work around circular dependency issues
self._whylabs_session_api = Lazy(self.__create_session_api)
self._user_guid = self._get_or_create_user_guid()
emit_usage("guest_session")

def __create_session_api(self) -> SessionsApi:
from whylogs.api.whylabs.session.whylabs_client_cache import ClientCacheConfig
Expand Down Expand Up @@ -255,15 +253,12 @@ def upload_batch_profile(self, profile: ResultSet) -> Union[UploadResult, NotSup

class ApiKeySession(Session):
def __init__(self, config: SessionConfig) -> None:
from whylogs.api.usage_stats import emit_usage

super().__init__(config)
self.api_key = config.get_api_key()
self.org_id = config.get_org_id()

# Using lazy initialization to work around circular dependency issues
self._whylabs_log_api = Lazy(partial(self.__create_log_api, config))
emit_usage("api_key_session")

def __create_log_api(self, config: SessionConfig) -> LogApi:
from whylogs.api.whylabs.session.whylabs_client_cache import ClientCacheConfig
Expand Down
2 changes: 0 additions & 2 deletions python/whylogs/api/writer/mlflow.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,6 @@

import mlflow

from whylogs.api.usage_stats import emit_usage
from whylogs.api.writer import Writer
from whylogs.api.writer.writer import _Writable
from whylogs.core.utils import deprecated_alias
Expand All @@ -18,7 +17,6 @@ def __init__(self) -> None:
self._file_dir = "whylogs"
self._file_name = None
self._end_run = True
emit_usage("mlflow_writer")

@deprecated_alias(profile="file")
def write(
Expand Down
2 changes: 0 additions & 2 deletions python/whylogs/api/writer/s3.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,6 @@
from botocore.client import BaseClient
from botocore.exceptions import ClientError

from whylogs.api.usage_stats import emit_usage
from whylogs.api.writer import Writer
from whylogs.api.writer.writer import _Writable
from whylogs.core.utils import deprecated_alias
Expand Down Expand Up @@ -63,7 +62,6 @@ def __init__(
self.base_prefix = base_prefix or "profile"
self.bucket_name = bucket_name or ""
self.object_name = object_name or None
emit_usage("s3_writer")

@deprecated_alias(profile="file")
def write(
Expand Down
4 changes: 0 additions & 4 deletions python/whylogs/core/constraints/__init__.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,3 @@
from whylogs.api.usage_stats import emit_usage

from .metric_constraints import (
Constraints,
ConstraintsBuilder,
Expand All @@ -19,5 +17,3 @@
MetricsSelector,
PrefixCondition,
]

emit_usage("metric_constraints")
4 changes: 0 additions & 4 deletions python/whylogs/core/constraints/factories/__init__.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,3 @@
from whylogs.api.usage_stats import emit_usage

from .cardinality_metrics import distinct_number_in_range
from .condition_counts import (
condition_count_below,
Expand Down Expand Up @@ -65,5 +63,3 @@
condition_count_below,
column_is_probably_unique,
]

emit_usage("constraints_factories")
3 changes: 0 additions & 3 deletions python/whylogs/core/segmentation_partition.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,6 @@
from dataclasses import dataclass, field
from typing import Callable, List, Mapping, Optional

from whylogs.api.usage_stats import emit_usage
from whylogs.core.projectors import FieldProjector

logger = logging.getLogger(__name__)
Expand Down Expand Up @@ -81,6 +80,4 @@ def __hash__(self):


def segment_on_column(column_name: str) -> Mapping[str, SegmentationPartition]:
emit_usage("segment_on_column")

return {column_name: SegmentationPartition(name=column_name, mapper=ColumnMapperFunction(col_names=[column_name]))}
Loading

0 comments on commit 5f45d5f

Please sign in to comment.