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

@@ -67,7 +67,7 @@ PythonEvent::process()
     }
 }
 
-Event *
+CountedDrainEvent *
 createCountedDrain()
 {
     return new CountedDrainEvent();

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__