code: def Wstar(self,value):
          
          if value is None:
              self._use_to_predict = False
          else:
              assert value.shape[1]==self._K, 'Dimension mismatch'
              self._use_to_predict = True
          self._Wstar = value
          self.clear_cache('predict')
docs: set fixed effect design for predictions