fix: force types in streams asset (#11)

**Motivation:** Force types for fields which were not specific enough
llamaxyz · Aug 10, 2023 · e7ebeff · e7ebeff
1 parent 2b7371a
commit e7ebeff
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/aave_data/assets/financials/data_lake.py b/aave_data/assets/financials/data_lake.py
@@ -1710,6 +1710,10 @@ def streaming_payments_state(context, block_numbers_by_day):
     streams['unvested'] = streams.deposit - streams.vested
     streams['unclaimed'] = streams.vested - streams.claims
 
+    # fix types
+    float_cols = ['stream_rate','claims_raw','deposit','claims','current_time','vested_proportion','vested','unvested','unclaimed']
+    for col in float_cols:
+        streams[col] = streams[col].astype(float)
 
     streams = streams.drop(columns=['deposit_raw', 'claims_raw', 'vested_proportion', 'current_time'])
     # streams = streams.drop(columns=['deposit_raw', 'claims_raw'])