diff --git a/src/litdata/utilities/encryption.py b/src/litdata/utilities/encryption.py index 35be895f..74e58fa5 100644 --- a/src/litdata/utilities/encryption.py +++ b/src/litdata/utilities/encryption.py @@ -22,7 +22,9 @@ def decrypt(self, data: bytes) -> bytes: class FernetEncryption(Encryption): """Encryption for the Fernet package. + Adapted from: https://cryptography.io/en/latest/fernet/ + """ def __init__(self, passsword: str) -> None: @@ -50,7 +52,9 @@ def _derive_key(self, password: str) -> bytes: class RSAEncryption: """Encryption for the RSA package. + Adapted from: https://cryptography.io/en/latest/hazmat/primitives/asymmetric/rsa/ + """ def __init__(self, private_key_path: str = None, public_key_path: str = None) -> None: