HSSM Conda Install Problem

[martamigo]

Hello!,

I get no errors when I install HSSM using "conda install -c conda-forge hssm". In fact, I'm even able to import it in my python environment and use it to run veeery simple models. As soon as I try to add a regressor to my model, even when I run the "Getting Started" code, I get the following error when I try to .sample():

Traceback (most recent call last):
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/vm.py", line 1227, in make_all
node.op.make_thunk(node, storage_map, compute_map, [], impl=impl)
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/op.py", line 119, in make_thunk
return self.make_c_thunk(node, storage_map, compute_map, no_recycling)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/op.py", line 84, in make_c_thunk
outputs = cl.make_thunk(
^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/basic.py", line 1182, in make_thunk
cthunk, module, in_storage, out_storage, error_storage = self.compile(
^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/basic.py", line 1103, in compile
thunk, module = self.cthunk_factory(
^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/basic.py", line 1627, in cthunk_factory
module = cache.module_from_key(key=key, lnk=self)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/cmodule.py", line 1255, in module_from_key
module = lnk.compile_cmodule(location)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/basic.py", line 1528, in compile_cmodule
module = c_compiler.compile_str(
^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/cmodule.py", line 2666, in compile_str
return dlimport(lib_filename)
^^^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/cmodule.py", line 351, in dlimport
rval = import(module_name, {}, {}, [module_name])
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ImportError: dlopen(/Users/martamigocortes/.pytensor/compiledir_macOS-10.16-x86_64-i386-64bit-i386-3.11.9-64/tmpbh31f8ig/me74e37e9c42976fda662dc2a17d4af4617dce1e2c92b41a904b6ced701075905.so, 0x0002): symbol not found in flat namespace '_mkl_dft_dfticommitdescriptor'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "", line 1, in
File "/opt/anaconda3/lib/python3.11/site-packages/hssm/hssm.py", line 568, in sample
self._inference_obj = self.model.fit(
^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/bambi/models.py", line 333, in fit
return self.backend.run(
^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/bambi/backend/pymc.py", line 96, in run
result = self._run_mcmc(
^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/bambi/backend/pymc.py", line 172, in _run_mcmc
idata = pm.sample(
^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pymc/sampling/mcmc.py", line 716, in sample
step = assign_step_methods(model, step, methods=pm.STEP_METHODS, step_kwargs=kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pymc/sampling/mcmc.py", line 237, in assign_step_methods
return instantiate_steppers(model, steps, selected_steps, step_kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pymc/sampling/mcmc.py", line 138, in instantiate_steppers
step = step_class(vars=vars, model=model, **args)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pymc/step_methods/hmc/nuts.py", line 180, in init
super().init(vars, **kwargs)
File "/opt/anaconda3/lib/python3.11/site-packages/pymc/step_methods/hmc/base_hmc.py", line 109, in init
super().init(vars, blocked=blocked, model=self._model, dtype=dtype, **pytensor_kwargs)
File "/opt/anaconda3/lib/python3.11/site-packages/pymc/step_methods/arraystep.py", line 163, in init
func = model.logp_dlogp_function(vars, dtype=dtype, **pytensor_kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pymc/model/core.py", line 626, in logp_dlogp_function
return ValueGradFunction(costs, grad_vars, extra_vars_and_values, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pymc/model/core.py", line 333, in init
self._pytensor_function = compile_pymc(inputs, outputs, givens=givens, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pymc/pytensorf.py", line 1039, in compile_pymc
pytensor_function = pytensor.function(
^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/compile/function/init.py", line 318, in function
fn = pfunc(
^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/compile/function/pfunc.py", line 465, in pfunc
return orig_function(
^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/compile/function/types.py", line 1762, in orig_function
fn = m.create(defaults)
^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/compile/function/types.py", line 1654, in create
_fn, _i, _o = self.linker.make_thunk(
^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/basic.py", line 245, in make_thunk
return self.make_all(
^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/vm.py", line 1236, in make_all
raise_with_op(fgraph, node)
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/utils.py", line 524, in raise_with_op
raise exc_value.with_traceback(exc_trace)
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/vm.py", line 1227, in make_all
node.op.make_thunk(node, storage_map, compute_map, [], impl=impl)
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/op.py", line 119, in make_thunk
return self.make_c_thunk(node, storage_map, compute_map, no_recycling)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/op.py", line 84, in make_c_thunk
outputs = cl.make_thunk(
^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/basic.py", line 1182, in make_thunk
cthunk, module, in_storage, out_storage, error_storage = self.compile(
^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/basic.py", line 1103, in compile
thunk, module = self.cthunk_factory(
^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/basic.py", line 1627, in cthunk_factory
module = cache.module_from_key(key=key, lnk=self)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/cmodule.py", line 1255, in module_from_key
module = lnk.compile_cmodule(location)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/basic.py", line 1528, in compile_cmodule
module = c_compiler.compile_str(
^^^^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/cmodule.py", line 2666, in compile_str
return dlimport(lib_filename)
^^^^^^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/lib/python3.11/site-packages/pytensor/link/c/cmodule.py", line 351, in dlimport
rval = import(module_name, {}, {}, [module_name])
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ImportError: dlopen(/Users/martamigocortes/.pytensor/compiledir_macOS-10.16-x86_64-i386-64bit-i386-3.11.9-64/tmpbh31f8ig/me74e37e9c42976fda662dc2a17d4af4617dce1e2c92b41a904b6ced701075905.so, 0x0002): symbol not found in flat namespace '_mkl_dft_dfticommitdescriptor'
Apply node that caused the error: CGemv{inplace}(AllocEmpty{dtype='float64'}.0, 1.0, ExpandDims{axis=0}.0, [ 0.655767 ... .75188068], 0.0)
Toposort index: 50
Inputs types: [TensorType(float64, shape=(1,)), TensorType(float64, shape=()), TensorType(float64, shape=(1, 3988)), TensorType(float64, shape=(3988,)), TensorType(float64, shape=())]

Has anyone gotten this error before? Any advice? <3

[digicosmos86]

@martamigo,

hmmm this is weird. Not sure exactly what is going on here, but it seems that it has something to do with MKL. Are you running the code on an Intel Mac?

There's a few things that might be helpful:

1. Run the code with `hssm.set_floatX("float32")
2. Restart the kernel and run the code again, if you are using Jupyter Notebook/Lab
3. If you are running on Intel Mac, you probably need to install a compatible MKL library, and if you are running on an ARM Mac, your environment is set up against an Intel Mac.