empty multidimensional numpy arrays causing error

[JW-swansea]

This section of code seems to have excluded something, numpy arrays can have a shape (n,0) in addition to (n,) (n,1) (0) etc.

timesfm/src/timesfm/xreg_lib.py

Lines 43 to 49 in 02bc2f2

	if x.ndim == 1:
	(i,) = x.shape
	di = 2**math.ceil(math.log2(i)) - i
	return jnp.pad(x, ((0, di),), mode="constant", constant_values=0.0)
	elif x.ndim == 2:
	i, j = x.shape
	di = 2**math.ceil(math.log2(i)) - i

This means that it's possible to have an array shape that registers as two dimensional, and still trips up log i with a log of zero problem.

Now arrays of the form (n,0) should have no entries, so there's obviously another problem I have to trace, but this is something I can report at least; I believe you need an extra case to account for empty but multi-dimensional arrays.