diff --git a/python/dp_accounting/dp_accounting/BUILD.bazel b/python/dp_accounting/dp_accounting/BUILD.bazel index 5a45d0aa..e5033076 100644 --- a/python/dp_accounting/dp_accounting/BUILD.bazel +++ b/python/dp_accounting/dp_accounting/BUILD.bazel @@ -33,6 +33,7 @@ py_library( ":dp_event_builder", ":mechanism_calibration", ":privacy_accountant", + ":utilities", "//dp_accounting/pld", "//dp_accounting/rdp", ], @@ -130,3 +131,27 @@ py_library( "@accounting_py_pip_deps//absl_py", ], ) + +py_library( + name = "utilities", + srcs = ["utilities.py"], + srcs_version = "PY3", + deps = [ + "@accounting_py_pip_deps//numpy", + "@accounting_py_pip_deps//scipy", + ], +) + +py_test( + name = "utilities_test", + srcs = ["utilities_test.py"], + python_version = "PY3", + srcs_version = "PY3", + deps = [ + ":dp_event", + ":utilities", + "//dp_accounting/pld:pld_privacy_accountant", + "@accounting_py_pip_deps//absl_py", + "@accounting_py_pip_deps//numpy", + ], +) diff --git a/python/dp_accounting/dp_accounting/__init__.py b/python/dp_accounting/dp_accounting/__init__.py index 113a8215..7734cacc 100644 --- a/python/dp_accounting/dp_accounting/__init__.py +++ b/python/dp_accounting/dp_accounting/__init__.py @@ -20,6 +20,7 @@ from dp_accounting import pld from dp_accounting import privacy_accountant from dp_accounting import rdp +from dp_accounting import utilities from dp_accounting.dp_event import ComposedDpEvent from dp_accounting.dp_event import DpEvent from dp_accounting.dp_event import GaussianDpEvent