Checkpointing: createCountedDrain function, it was only returning an Event, which...

Signed-off By: Ali Saidi

diff --git a/src/python/swig/pyevent.cc b/src/python/swig/pyevent.cc
index bf993bc3e683b2098fd72c27a56155ee486eb9bd..a201e01856db2fd1cf20292784224b812084e482 100644 (file)
--- 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 c6d6ac9939e43ce2e1e1c56a9338b2be08216f89..22b562de1b23adeaa36abdf5704858e002884d96 100644 (file)
--- 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__