From 54b5f6f8eeb2598f18e2ef5e94ebe95c08d82395 Mon Sep 17 00:00:00 2001 From: Eduardo Blancas Date: Tue, 17 Sep 2024 23:44:16 -0600 Subject: [PATCH] remove telemetry --- CHANGELOG.md | 2 ++ doc/community/support.md | 4 ---- src/ploomber_engine/_telemetry.py | 14 -------------- src/ploomber_engine/engine.py | 4 ---- src/ploomber_engine/execute.py | 4 ---- src/ploomber_engine/profiling.py | 3 +-- src/ploomber_engine/tracking/tracking.py | 2 -- tests/test_tracking.py | 16 ---------------- 8 files changed, 3 insertions(+), 46 deletions(-) delete mode 100644 src/ploomber_engine/_telemetry.py diff --git a/CHANGELOG.md b/CHANGELOG.md index 9b874ad..a1f9cd4 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -2,6 +2,8 @@ ## 0.0.33dev +* [Feature] Remove telemetry + ## 0.0.32 (2024-02-07) * [Feature] Add benchmarking CLI `python -m ploomber_engine.benchmark` diff --git a/doc/community/support.md b/doc/community/support.md index b815854..8c28c4f 100644 --- a/doc/community/support.md +++ b/doc/community/support.md @@ -1,7 +1,3 @@ # Support For support, feature requests, and product updates: [join our community](https://ploomber.io/community) or follow us on [Twitter](https://twitter.com/ploomber)/[LinkedIn](https://www.linkedin.com/company/ploomber/). - -# Telemetry - -We collect (optional) anonymous statistics to understand and improve usage. For details, [see here](https://docs.ploomber.io/en/latest/community/user-stats.html). diff --git a/src/ploomber_engine/_telemetry.py b/src/ploomber_engine/_telemetry.py deleted file mode 100644 index d0e67d2..0000000 --- a/src/ploomber_engine/_telemetry.py +++ /dev/null @@ -1,14 +0,0 @@ -try: - from importlib.metadata import version -except ModuleNotFoundError: - from importlib_metadata import version - - -from ploomber_core.telemetry.telemetry import Telemetry - - -telemetry = Telemetry( - api_key="phc_P9SpSeypyPwxrMdFn2edOOEooQioF2axppyEeDwtMSP", - package_name="ploomber-engine", - version=version("ploomber-engine"), -) diff --git a/src/ploomber_engine/engine.py b/src/ploomber_engine/engine.py index 4b49aaa..7ea91c3 100644 --- a/src/ploomber_engine/engine.py +++ b/src/ploomber_engine/engine.py @@ -9,14 +9,12 @@ from ploomber_engine.papermill import PapermillPloomberNotebookClient from ploomber_engine.ipython import PloomberManagedClient -from ploomber_engine._telemetry import telemetry class DebugEngine(Engine): """An engine that starts a debugging session once the notebook fails""" @classmethod - @telemetry.log_call("debug-execute-managed-nb") def execute_managed_notebook( cls, nb_man, @@ -57,7 +55,6 @@ class DebugLaterEngine(Engine): """An engine that stores the traceback object for later debugging""" @classmethod - @telemetry.log_call("debuglater-execute-managed-nb") def execute_managed_notebook( cls, nb_man, @@ -118,7 +115,6 @@ class ProfilingEngine(Engine): """ @classmethod - @telemetry.log_call("embedded-execute-managed-nb") def execute_managed_notebook( cls, nb_man, diff --git a/src/ploomber_engine/execute.py b/src/ploomber_engine/execute.py index b92f268..7644a52 100644 --- a/src/ploomber_engine/execute.py +++ b/src/ploomber_engine/execute.py @@ -12,12 +12,8 @@ from ploomber_engine.ipython import PloomberClient from ploomber_engine import profiling from ploomber_engine import _util -from ploomber_engine._telemetry import telemetry -@telemetry.log_call( - log_args=True, ignore_args={"input_path", "output_path", "parameters"} -) def execute_notebook( input_path, output_path, diff --git a/src/ploomber_engine/profiling.py b/src/ploomber_engine/profiling.py index 49d1af9..d90dd08 100644 --- a/src/ploomber_engine/profiling.py +++ b/src/ploomber_engine/profiling.py @@ -7,7 +7,7 @@ from ploomber_engine.ipython import PloomberClient from ploomber_engine._util import recursive_update -from ploomber_engine._telemetry import telemetry + try: import psutil @@ -43,7 +43,6 @@ def hook_cell_post(self, cell): recursive_update(cell.metadata, metadata) -@telemetry.log_call("memory-profile") def memory_profile(path, output): path = Path(path) target = path.with_name(path.stem + "-memory-usage.png") diff --git a/src/ploomber_engine/tracking/tracking.py b/src/ploomber_engine/tracking/tracking.py index ec43837..3cb672f 100644 --- a/src/ploomber_engine/tracking/tracking.py +++ b/src/ploomber_engine/tracking/tracking.py @@ -16,7 +16,6 @@ from ploomber_engine.ipython import PloomberClient, add_to_sys_path from ploomber_engine.tracking.io import _process_content_data -from ploomber_engine._telemetry import telemetry from ploomber_engine._util import find_cell_with_parameters_comment try: @@ -161,7 +160,6 @@ def _parse_cli_parameters(parameters): return {k: _parse_param(v) for k, v in pairs} -@telemetry.log_call("track-execution") def track_execution(filename, parameters=None, database="experiments.db", quiet=False): """ Execute a script or notebook and write outputs to a SQLite database diff --git a/tests/test_tracking.py b/tests/test_tracking.py index e09fe55..a2c4d55 100644 --- a/tests/test_tracking.py +++ b/tests/test_tracking.py @@ -6,7 +6,6 @@ from ploomber_engine.tracking.tracking import _parse_cli_parameters, extract_name from ploomber_engine.tracking import track_execution from sklearn_evaluation import SQLiteTracker -from ploomber_engine import _telemetry @pytest.mark.parametrize( @@ -297,18 +296,3 @@ def sum_(args): ) def test_extract_name(source, expected): assert extract_name(source) == expected - - -def test_tracking_import_telemetry(tmp_empty, monkeypatch): - Path("functions.py").write_text("""x = 1""") - mock = Mock() - monkeypatch.setattr(_telemetry.telemetry, "log_api", mock) - - track_execution( - filename="functions.py", - parameters=dict(a=1, b=2), - database="exps.db", - quiet=True, - ) - - assert mock.call_count == 1