From 2cc9cab17fb9814bcb48bdf7e5b284ebacff540d Mon Sep 17 00:00:00 2001 From: Richard Frank Date: Fri, 20 Dec 2013 17:11:53 -0500 Subject: [PATCH] MAINT: Initialize Portfolio object with default values for attributes instead of no attributes at all --- zipline/protocol.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/zipline/protocol.py b/zipline/protocol.py index 1f1e3ba2..d2a65f16 100644 --- a/zipline/protocol.py +++ b/zipline/protocol.py @@ -67,9 +67,17 @@ class Order(Event): class Portfolio(object): - def __init__(self, initial_values=None): - if initial_values: - self.__dict__ = initial_values + def __init__(self): + self.capital_used = 0.0 + self.starting_cash = 0.0 + self.portfolio_value = 0.0 + self.pnl = 0.0 + self.returns = 0.0 + self.cash = 0.0 + self.positions = Positions() + self.start_date = None + self.positions_value = 0.0 + self.portfolio_value = 0.0 def __getitem__(self, key): return self.__dict__[key]