RuntimeError: KeyNotFoundException while rearranging houses from ProcThor

Hi @mattdeitke,

I'm trying to use ProcThor-10k dataset to generate some new scenes with rearrangement for the rearrangement-challenge.
While generating, I'm getting this error for some of the houses from ProcThor. So it's just for some of them - not all. I'm confused about which key is missing in the dictionary and how can I fix it.
I'm using only 1-room houses from ProcThor, if it helps. I did some experiments using 4.3.0 version of Ai2Thor and custom 0+391b3fae4d4cc026f1522e5acf60953560235971 as in colab for ProcThor, but seems the error is always same for the same scenes.
Will appreciate any help!

`[08/07 07:00:48 ERROR:] RearrangeProcTHORDatagenWorker 0: Exception when processing rearrange 4 {'scene': 'train_14', 'stage': 'train', 'seed': 24197580, 'reuse_i': 0, 'obj_name_to_avoid_positions': None, 'limits': {'room_openables': {'room|4': ['Fridge_5_0']}, 'max_pickupable': 20, 'max_receps': {'room|4': 10}}, 'scene_reuse_count': 2} [multiprocessing_utils.py: 62]
Traceback (most recent call last):
File "/dstore/home/mkhan/ai2thor-rearrangement-procthor/utils/multiprocessing_utils.py", line 56, in __init__
res = self.work(job[0], job[2])
File "datagen/procthor_datagen/datagen_runner.py", line 1185, in work
data = generate_rearrangement_for_scene(
File "datagen/procthor_datagen/datagen_runner.py", line 788, in generate_rearrangement_for_scene
if not env.procthor_reset(scene_name=scene, force_reset=True):
File "/dstore/home/mkhan/ai2thor-rearrangement-procthor/rearrange/procthor_rearrange/environment.py", line 577, in procthor_reset
self.controller.step(
File "/home/mkhan/anaconda3/envs/thor-rearrange/lib/python3.8/site-packages/ai2thor/controller.py", line 985, in step
raise RuntimeError(
RuntimeError: KeyNotFoundException: The given key was not present in the dictionary.. trace: at System.Collections.Generic.Dictionary`2[TKey,TValue].get_Item (TKey key) [0x0001e] in <695d1cc93cca45069c528c15c9fdd749>:0
at Thor.Procedural.AssetMap`1[T].getAsset (System.String name) [0x00000] in <6afd8f78be764eeba7be30f178fa1cb8>:0
at Thor.Procedural.ProceduralTools.CreateHouse (Thor.Procedural.Data.ProceduralHouse house, Thor.Procedural.AssetMap`1[T] materialDb, System.Nullable`1[T] position) [0x0035a] in <6afd8f78be764eeba7be30f178fa1cb8>:0
at UnityStandardAssets.Characters.FirstPerson.BaseFPSAgentController.CreateHouse (Thor.Procedural.Data.ProceduralHouse house) [0x000e1] in <6afd8f78be764eeba7be30f178fa1cb8>:0
at (wrapper managed-to-native) System.Reflection.MonoMethod.InternalInvoke(System.Reflection.MonoMethod,object,object[],System.Exception&)
at System.Reflection.MonoMethod.Invoke (System.Object obj, System.Reflection.BindingFlags invokeAttr, System.Reflection.Binder binder, System.Object[] parameters, System.Globalization.CultureInfo culture) [0x00032] in <695d1cc93cca45069c528c15c9fdd749>:0
`