driver.py

from hamilton import driver, base
from hamilton import log_setup


dr = (
  driver
  .Builder()
  .with_modules(base_dates, probabilities)
  .with_adapters(base.PandasDataFrameResult())
  .build()
)

delivery_probabilities = dr.execute(
    ["possible_dates", "probability_before_date", "probability_on_date"], 
    inputs={"start_date" : PREGNANCY_START_DATE, "current_date": datetime.datetime.now()}
)