Modified sys.path in gym_iw_agent.py so python can find the compiled module after pip installing gym_iw

Author: fteicht

gym_iw/gym_iw_agent.py

@@ -1,7 +1,10 @@
 import argparse
 import sys
+import os
 
-from gym_iw import __GymIwAgent
+sys.path.append(os.path.dirname(os.path.realpath(__file__)))
+
+from __gym_iw import __GymIwAgent
 
 import gym
 from gym import wrappers, logger

src/CMakeLists.txt

@@ -1 +1 @@
-PYBIND11_ADD_MODULE(gym_iw gym_spaces.cc gym_iw.cc logger.cc)
+PYBIND11_ADD_MODULE(__gym_iw gym_spaces.cc gym_iw.cc logger.cc)

src/gym_iw.cc

@@ -5,7 +5,7 @@
 
 namespace py = pybind11;
 
-PYBIND11_MODULE(gym_iw, giw) {
+PYBIND11_MODULE(__gym_iw, giw) {
     py::class_<GymProxy> gym_iw_agent(giw, "__GymIwAgent");
         gym_iw_agent
             .def(py::init<const py::object&, const std::string&, const std::string&,