Skip to content

Commit

Permalink
Fix formatting errors
Browse files Browse the repository at this point in the history
  • Loading branch information
fstp committed Aug 28, 2024
1 parent 3b01960 commit 46a8938
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 2 deletions.
3 changes: 2 additions & 1 deletion src/zipline/pipeline/loaders/earnings_estimates.py
Original file line number Diff line number Diff line change
Expand Up @@ -702,7 +702,8 @@ def get_last_data_per_qtr(
ffill_across_cols(last_per_qtr, columns, self.name_map)
# Stack quarter and sid into the index.
stacked_last_per_qtr = last_per_qtr.stack(
[SID_FIELD_NAME, NORMALIZED_QUARTERS], future_stack=True,
[SID_FIELD_NAME, NORMALIZED_QUARTERS],
future_stack=True,
)
# Set date index name for ease of reference
stacked_last_per_qtr.index.set_names(
Expand Down
5 changes: 4 additions & 1 deletion tests/pipeline/test_international_markets.py
Original file line number Diff line number Diff line change
Expand Up @@ -128,7 +128,10 @@ def init_class_fixtures(cls):

bar_data = cls.daily_bar_data[name]
df = (
pd.concat(bar_data, keys=bar_data.keys()).stack(future_stack=True).unstack(0).swaplevel()
pd.concat(bar_data, keys=bar_data.keys())
.stack(future_stack=True)
.unstack(0)
.swaplevel()
)
frames = {
field: frame.reset_index(level=0, drop=True)
Expand Down

0 comments on commit 46a8938

Please sign in to comment.