Skip to content

Commit 090b504

Browse files
committed
fix(vtk-module): fix import for custom vtk module
1 parent cc5819d commit 090b504

File tree

1 file changed

+7
-1
lines changed

1 file changed

+7
-1
lines changed

src/trame_vtk/modules/vtk/serializers/synchronization_context.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,14 @@
22
import logging
33
import time
44
import zipfile
5+
import os
6+
import importlib
7+
import sys
58

6-
from vtkmodules.vtkCommonCore import vtkTypeUInt32Array, vtkFloatArray, vtkDoubleArray
9+
vtk_module_name = os.environ.get("VTK_MODULE_NAME", "vtkmodules")
10+
sys.modules["vtk_module"] = importlib.import_module(vtk_module_name)
11+
12+
from vtk_module.vtkCommonCore import vtkTypeUInt32Array, vtkFloatArray, vtkDoubleArray
713

814
from .utils import base64_encode, wrap_id
915

0 commit comments

Comments
 (0)