Theano tmp module 'tmp_...' not found

[avivajpeyi]

115/~3K analyses had a theano error that looks like:

planet_transit_model, params = build_planet_transit_model(tic_entry)
model_varnames = get_untransformed_varnames(planet_transit_model)
test_model(planet_transit_model)
---------------------------------------------------------------------------
ModuleNotFoundError                       Traceback (most recent call last)
<ipython-input-1-797d677cee95> in <module>
----> 1 planet_transit_model, params = build_planet_transit_model(tic_entry)
      2 model_varnames = get_untransformed_varnames(planet_transit_model)
      3 test_model(planet_transit_model)

<ipython-input-1-44cb72db158f> in build_planet_transit_model(tic_entry)
    102                     upper=planet.tmax + planet.duration_max,
    103                 )
--> 104                 tmax_prior = tmax_norm(
    105                     name=f"{TIME_END}_{planet.index}",
    106                     mu=planet.tmax,

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/pymc3/distributions/bound.py in __call__(self, name, *args, **kwargs)
    293         transform = kwargs.pop("transform", "infer")
    294         if issubclass(self.distribution, Continuous):
--> 295             return _ContinuousBounded(
    296                 name, self.distribution, self.lower, self.upper, transform, *args, **kwargs
    297             )

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/pymc3/distributions/distribution.py in __new__(cls, name, *args, **kwargs)
    120         else:
    121             dist = cls.dist(*args, **kwargs)
--> 122         return model.Var(name, dist, data, total_size, dims=dims)
    123 
    124     def __getnewargs__(self):

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/pymc3/model.py in Var(self, name, dist, data, total_size, dims)
   1140             else:
   1141                 with self:
-> 1142                     var = TransformedRV(
   1143                         name=name,
   1144                         distribution=dist,

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/pymc3/model.py in __init__(self, type, owner, index, name, distribution, model, transform, total_size)
   2011 
   2012             self.transformed = model.Var(
-> 2013                 transformed_name, transform.apply(distribution), total_size=total_size
   2014             )
   2015 

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/pymc3/distributions/transforms.py in apply(self, dist)
    124     def apply(self, dist):
    125         # avoid circular import
--> 126         return TransformedDistribution.dist(dist, self)
    127 
    128     def __str__(self):

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/pymc3/distributions/distribution.py in dist(cls, *args, **kwargs)
    128     def dist(cls, *args, **kwargs):
    129         dist = object.__new__(cls)
--> 130         dist.__init__(*args, **kwargs)
    131         return dist
    132 

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/pymc3/distributions/transforms.py in __init__(self, dist, transform, *args, **kwargs)
    148             arguments to Distribution"""
    149         forward = transform.forward
--> 150         testval = forward(dist.default())
    151 
    152         self.dist = dist

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/pymc3/distributions/transforms.py in forward(self, x)
    285     def forward(self, x):
    286         a, b = self.a, self.b
--> 287         return tt.log(x - a) - tt.log(b - x)
    288 
    289     def forward_val(self, x, point=None):

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/theano/tensor/var.py in __rsub__(self, other)
    199 
    200     def __rsub__(self, other):
--> 201         return theano.tensor.basic.sub(other, self)
    202 
    203     def __rmul__(self, other):

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/theano/graph/op.py in __call__(self, *inputs, **kwargs)
    251 
    252         if config.compute_test_value != "off":
--> 253             compute_test_value(node)
    254 
    255         if self.default_output is not None:

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/theano/graph/op.py in compute_test_value(node)
    124 
    125     # Create a thunk that performs the computation
--> 126     thunk = node.op.make_thunk(node, storage_map, compute_map, no_recycling=])
    127     thunk.inputs = storage_mapv] for v in node.inputs]
    128     thunk.outputs = storage_mapv] for v in node.outputs]

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/theano/graph/op.py in make_thunk(self, node, storage_map, compute_map, no_recycling, impl)
    632             )
    633             try:
--> 634                 return self.make_c_thunk(node, storage_map, compute_map, no_recycling)
    635             except (NotImplementedError, MethodNotDefined):
    636                 # We requested the c code, so don't catch the error.

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/theano/graph/op.py in make_c_thunk(self, node, storage_map, compute_map, no_recycling)
    598                 print(f"Disabling C code for {self} due to unsupported float16")
    599                 raise NotImplementedError("float16")
--> 600         outputs = cl.make_thunk(
    601             input_storage=node_input_storage, output_storage=node_output_storage
    602         )

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/theano/link/c/basic.py in make_thunk(self, input_storage, output_storage, storage_map)
   1201         """
   1202         init_tasks, tasks = self.get_init_tasks()
-> 1203         cthunk, module, in_storage, out_storage, error_storage = self.__compile__(
   1204             input_storage, output_storage, storage_map
   1205         )

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/theano/link/c/basic.py in __compile__(self, input_storage, output_storage, storage_map)
   1136         input_storage = tuple(input_storage)
   1137         output_storage = tuple(output_storage)
-> 1138         thunk, module = self.cthunk_factory(
   1139             error_storage,
   1140             input_storage,

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/theano/link/c/basic.py in cthunk_factory(self, error_storage, in_storage, out_storage, storage_map)
   1632             for node in self.node_order:
   1633                 node.op.prepare_node(node, storage_map, None, "c")
-> 1634             module = get_module_cache().module_from_key(key=key, lnk=self)
   1635 
   1636         vars = self.inputs + self.outputs + self.orphans

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/theano/link/c/cmodule.py in module_from_key(self, key, lnk)
   1155         # Is the source code already in the cache?
   1156         module_hash = get_module_hash(src_code, key)
-> 1157         module = self._get_from_hash(module_hash, key)
   1158         if module is not None:
   1159             return module

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/theano/link/c/cmodule.py in _get_from_hash(self, module_hash, key)
   1055         if module_hash in self.module_hash_to_key_data:
   1056             key_data = self.module_hash_to_key_datamodule_hash]
-> 1057             module = self._get_from_key(None, key_data)
   1058             with lock_ctx():
   1059                 try:

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/theano/link/c/cmodule.py in _get_from_key(self, key, key_data)
   1050         if name is None:
   1051             return None
-> 1052         return self._get_module(name)
   1053 
   1054     def _get_from_hash(self, module_hash, key):

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/theano/link/c/cmodule.py in _get_module(self, name)
    698         if name not in self.module_from_name:
    699             _logger.debug(f"loading name {name}")
--> 700             self.module_from_namename] = dlimport(name)
    701             self.stats1] += 1
    702         else:

/fred/oz200/avajpeyi/envs/tess/lib/python3.8/site-packages/theano/link/c/cmodule.py in dlimport(fullpath, suffix)
    292         with warnings.catch_warnings():
    293             warnings.filterwarnings("ignore", message="numpy.ndarray size changed")
--> 294             rval = __import__(module_name, {}, {}, module_name])
    295         t1 = time.time()
    296         import_time += t1 - t0

ModuleNotFoundError: No module named 'tmp_s9tpyf8'

time: 2min 11s (started: 2023-07-20 16:35:00 +10:00)

[avivajpeyi]

Fixed with 121b4ce47d32b23764e287b38992abacd8c2f3c3