From a2c4babd93cd0b4a025fd64bb06454f3f0e99e99 Mon Sep 17 00:00:00 2001
From: fawce <fawce@quantopian.com>
Date: Fri, 16 Mar 2012 17:54:34 -0400
Subject: [PATCH] hooking perf to the on_done callback.

---
 zipline/finance/trading.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/zipline/finance/trading.py b/zipline/finance/trading.py
index 6c3d1318..e6119a8e 100644
--- a/zipline/finance/trading.py
+++ b/zipline/finance/trading.py
@@ -29,6 +29,8 @@ class TradeSimulationClient(qmsg.Component):
         )
         
         self.perf = perf.PerformanceTracker(self.trading_environment)
+        self.on_done = self.perf.handle_simulation_end
+        
     
     @property
     def get_id(self):