Traceback (most recent call last):
  File "C:/Users/ninadmore/IdeaProjects/tensorflow/python/cards/mendikot/play/individual_game/ai_game.py", line 24, in <module>
    print(play_ai_game())
  File "C:/Users/ninadmore/IdeaProjects/tensorflow/python/cards/mendikot/play/individual_game/ai_game.py", line 17, in play_ai_game
    board_controller.play_board()
  File "C:\Users\ninadmore\IdeaProjects\tensorflow\python\cards\mendikot\board\board_controller.py", line 17, in play_board
    self.step()
  File "C:\Users\ninadmore\IdeaProjects\tensorflow\python\cards\mendikot\board\board_controller.py", line 27, in step
    card = player.get_card_to_play(self.board.joint_observations)
  File "C:\Users\ninadmore\IdeaProjects\tensorflow\python\cards\mendikot\player\ai_player.py", line 97, in get_card_to_play
    action = self.agent.collect_policy.action(current_time_step)
  File "C:\Users\ninadmore\AppData\Local\Programs\Python\Python36\lib\site-packages\tf_agents\policies\tf_policy.py", line 324, in action
    step = action_fn(time_step=time_step, policy_state=policy_state, seed=seed)
  File "C:\Users\ninadmore\AppData\Local\Programs\Python\Python36\lib\site-packages\tf_agents\utils\common.py", line 185, in with_check_resource_vars
    return fn(*fn_args, **fn_kwargs)
  File "C:\Users\ninadmore\AppData\Local\Programs\Python\Python36\lib\site-packages\tf_agents\policies\epsilon_greedy_policy.py", line 107, in _action
    greedy_action = self._greedy_policy.action(time_step, policy_state)
  File "C:\Users\ninadmore\AppData\Local\Programs\Python\Python36\lib\site-packages\tf_agents\policies\tf_policy.py", line 324, in action
    step = action_fn(time_step=time_step, policy_state=policy_state, seed=seed)
  File "C:\Users\ninadmore\AppData\Local\Programs\Python\Python36\lib\site-packages\tf_agents\utils\common.py", line 185, in with_check_resource_vars
    return fn(*fn_args, **fn_kwargs)
  File "C:\Users\ninadmore\AppData\Local\Programs\Python\Python36\lib\site-packages\tf_agents\policies\tf_policy.py", line 560, in _action
    distribution_step = self._distribution(time_step, policy_state)
  File "C:\Users\ninadmore\AppData\Local\Programs\Python\Python36\lib\site-packages\tf_agents\policies\greedy_policy.py", line 81, in _distribution
    time_step, policy_state)
  File "C:\Users\ninadmore\AppData\Local\Programs\Python\Python36\lib\site-packages\tf_agents\policies\tf_policy.py", line 403, in distribution
    step = self._distribution(time_step=time_step, policy_state=policy_state)
  File "C:\Users\ninadmore\AppData\Local\Programs\Python\Python36\lib\site-packages\tf_agents\policies\q_policy.py", line 155, in _distribution
    step_type=time_step.step_type)
  File "C:\Users\ninadmore\AppData\Local\Programs\Python\Python36\lib\site-packages\tf_agents\networks\network.py", line 413, in __call__
    outputs, new_state = super(Network, self).__call__(**normalized_kwargs)
  File "C:\Users\ninadmore\AppData\Roaming\Python\Python36\site-packages\tensorflow\python\keras\engine\base_layer.py", line 985, in __call__
    outputs = call_fn(inputs, *args, **kwargs)
  File "C:\Users\ninadmore\AppData\Local\Programs\Python\Python36\lib\site-packages\tf_agents\networks\q_network.py", line 148, in call
    q_value = self._q_value_layer(state, training=training)
  File "C:\Users\ninadmore\AppData\Roaming\Python\Python36\site-packages\tensorflow\python\keras\engine\base_layer.py", line 985, in __call__
    outputs = call_fn(inputs, *args, **kwargs)
  File "C:\Users\ninadmore\AppData\Roaming\Python\Python36\site-packages\tensorflow\python\keras\layers\core.py", line 1198, in call
    dtype=self._compute_dtype_object)
  File "C:\Users\ninadmore\AppData\Roaming\Python\Python36\site-packages\tensorflow\python\keras\layers\ops\core.py", line 64, in dense
    outputs = nn_ops.bias_add(outputs, bias)
  File "C:\Users\ninadmore\AppData\Roaming\Python\Python36\site-packages\tensorflow\python\util\dispatch.py", line 201, in wrapper
    return target(*args, **kwargs)
  File "C:\Users\ninadmore\AppData\Roaming\Python\Python36\site-packages\tensorflow\python\ops\nn_ops.py", line 3366, in bias_add
    value, bias, data_format=data_format, name=name)
  File "C:\Users\ninadmore\AppData\Roaming\Python\Python36\site-packages\tensorflow\python\ops\gen_nn_ops.py", line 681, in bias_add
    _ops.raise_from_not_ok_status(e, name)
  File "C:\Users\ninadmore\AppData\Roaming\Python\Python36\site-packages\tensorflow\python\framework\ops.py", line 6843, in raise_from_not_ok_status
    six.raise_from(core._status_to_exception(e.code, message), None)
  File "<string>", line 3, in raise_from
tensorflow.python.framework.errors_impl.InvalidArgumentError: Input tensor must be at least 2D: [52] [Op:BiasAdd]