Bug and/or dead code in modular_network.py?

[sflc6]

In particular, see here: https://github.com/heronsystems/adeptRL/blob/2d8b0d3be9d0530f0df55d7f4b919177f8767388/adept/network/modular_network.py#L216-L238

In the following if-statement:

if dim in head_submodules:
    continue

type(dim) == int; however, keys of head_submodules are guaranteed to be str due to the line head_submodules[str(dim)] = submod. This suggests that the above if-statement will never continue. Thoughts on this?

[sflc6]

Perhaps we want dim = str(len(shape)) on line 220?