diff --git a/src/tinfoil/__init__.py b/src/tinfoil/__init__.py index 0317ccf..7827711 100644 --- a/src/tinfoil/__init__.py +++ b/src/tinfoil/__init__.py @@ -115,4 +115,4 @@ def NewSecureClient(enclave: str = "", repo: str = "tinfoilsh/confidential-model tf_client = SecureClient(enclave, repo, measurement) return _HTTPSecureClient(enclave, tf_client, api_key) -__all__ = ["TinfoilAI", "AsyncTinfoilAI", "NewSecureClient"] +__all__ = ["TinfoilAI", "AsyncTinfoilAI", "NewSecureClient", "SecureClient"]