From 0e87a3127262b2a9519f884c4d7ed676946315a3 Mon Sep 17 00:00:00 2001 From: Felipe Date: Wed, 21 Aug 2024 10:46:56 -0700 Subject: [PATCH] Fix it for Windows --- pyproject.toml | 1 - sdgym/benchmark.py | 5 ++--- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index e9d12fc4..91252595 100644 --- 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 index 186cf063..6c70a02c 100644 --- 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