code: def set_npartitions(self, npartitions=None):
          
          if npartitions is None:
              self._npartitions = cpu_count() * 2
          else:
              self._npartitions = npartitions
          return self
docs: Set the number of partitions to use for dask