afml/sample_weights/__init__.py

"""
Contains the code for implementing sample weights.
"""

if __name__ == "__main__" and not __package__:
    print("This file is a package initializer and is not meant to be run directly.")
    print("Use it via import, for example:")
    print(r"  .\.venv\Scripts\python.exe -c ""from afml.sample_weights import get_weights_by_return""")
    raise SystemExit(0)

from .attribution import get_weights_by_return, get_weights_by_time_decay
from .optimized_attribution import (
    get_weights_by_return_optimized,
    get_weights_by_time_decay_optimized,
)