Skip to content

Commit

Permalink
Add runtime to feature generation logs
Browse files Browse the repository at this point in the history
  • Loading branch information
bfhealy committed Dec 27, 2023
1 parent 5982a74 commit a2277cc
Showing 1 changed file with 5 additions and 0 deletions.
5 changes: 5 additions & 0 deletions tools/generate_features.py
Original file line number Diff line number Diff line change
Expand Up @@ -28,6 +28,7 @@
import json
from joblib import Parallel, delayed
from scipy.stats import circmean
import time


BASE_DIR = pathlib.Path(__file__).parent.parent.absolute()
Expand Down Expand Up @@ -507,6 +508,7 @@ def generate_features(
* - specified in config.yaml
"""
t0 = time.time()

# Get code version and current date/time for metadata
code_version = scope.__version__
Expand Down Expand Up @@ -1194,6 +1196,9 @@ def generate_features(
else:
print(f"Generated features for {len(feature_df)} sources.")

t1 = time.time()
print(f"Finished running in {t1 - t0} seconds.")

return feature_df


Expand Down

0 comments on commit a2277cc

Please sign in to comment.