Fix it for Windows

sdv-dev · Aug 21, 2024 · 0e87a31 · 0e87a31
1 parent 7cbd778
commit 0e87a31
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -49,7 +49,6 @@ dependencies = [
     'rdt>=1.12.1',
     'sdmetrics>=0.14.1',
     'sdv>=1.13.1',
-    'multiprocess>=0.70.16'
 ]
 
 [project.urls]

diff --git a/sdgym/benchmark.py b/sdgym/benchmark.py
@@ -2,6 +2,7 @@
 
 import concurrent
 import logging
+import multiprocessing
 import os
 import pickle
 import tracemalloc
@@ -10,6 +11,7 @@
 from pathlib import Path
 
 import boto3
+import cloudpickle
 import compress_pickle
 import numpy as np
 import pandas as pd
@@ -42,9 +44,6 @@
     get_synthesizers,
     used_memory,
 )
-import multiprocessing
-import cloudpickle
-import dill
 
 multiprocessing.set_start_method('spawn', force=True)
 multiprocessing.reduction.ForkingPickler.dumps = cloudpickle.dumps