clean all cpp files

Author: rtavenar

setup.py

@@ -33,8 +33,12 @@
 
 # clean cython output is clean is called
 if "clean" in sys.argv[1:]:
-    if os.path.isfile("ot/lp/emd_wrap.cpp"):
-        os.remove("ot/lp/emd_wrap.cpp")
+    for cpp_file in [
+        "ot/lp/emd_wrap.cpp",
+        "ot/partial/partial_cython.cpp",
+    ]:
+        if os.path.isfile(cpp_file):
+            os.remove(cpp_file)
 
 # add platform dependant optional compilation argument
 openmp_supported, flags = check_openmp_support()