Add msg

sdv-dev · May 16, 2024 · d68475f · d68475f
1 parent c2a1a6a
commit d68475f
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/sdv/single_table/base.py b/sdv/single_table/base.py
@@ -492,7 +492,14 @@ def load(cls, filepath):
                 The loaded synthesizer.
         """
         with open(filepath, 'rb') as f:
-            synthesizer = cloudpickle.load(f)
+            try:
+                synthesizer = cloudpickle.load(f)
+            except RuntimeError:
+                raise SamplingError(
+                    'This synthesizer was created on a machine with GPU but the current machine is' 
+                    ' CPU-only. This feature is currently unsupported. We recommend sampling on '
+                    'the same GPU-enabled machine.'
+                )
 
         check_synthesizer_version(synthesizer)
         check_sdv_versions_and_warn(synthesizer)