From cfa32d8de737847f04bfa6e5e89375c870c3869b Mon Sep 17 00:00:00 2001
From: Clint Smullen <cws3k@cs.virginia.edu>
Date: Mon, 27 Oct 2008 19:46:01 -0400
Subject: Checkpointing: createCountedDrain function, it was only returning an
 Event, which does not expose a setCount method to Python.

Signed-off By: Ali Saidi
---
 src/python/swig/pyevent.cc | 2 +-
 src/python/swig/pyevent.hh | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

(limited to 'src/python')

diff --git a/src/python/swig/pyevent.cc b/src/python/swig/pyevent.cc
index bf993bc3e..a201e0185 100644
--- a/src/python/swig/pyevent.cc
+++ b/src/python/swig/pyevent.cc
@@ -67,7 +67,7 @@ PythonEvent::process()
     }
 }
 
-Event *
+CountedDrainEvent *
 createCountedDrain()
 {
     return new CountedDrainEvent();
diff --git a/src/python/swig/pyevent.hh b/src/python/swig/pyevent.hh
index c6d6ac993..22b562de1 100644
--- a/src/python/swig/pyevent.hh
+++ b/src/python/swig/pyevent.hh
@@ -46,7 +46,7 @@ class PythonEvent : public Event
     virtual void process();
 };
 
-Event *createCountedDrain();
+CountedDrainEvent *createCountedDrain();
 void cleanupCountedDrain(Event *counted_drain);
 
 #endif // __PYTHON_SWIG_PYEVENT_HH__
-- 
cgit v1.2.3