use yield from (ruff UP028)

sunqm · Oct 16, 2023 · ceea353 · ceea353
1 parent c1ae259
commit ceea353
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 8 deletions.
diff --git a/pyscf/agf2/dfragf2.py b/pyscf/agf2/dfragf2.py
@@ -306,8 +306,7 @@ def prange(self, start=None, stop=None, step=None):
         if stop is None: stop = self.get_naoaux()
         if step is None: step = self.blockdim
 
-        for p0, p1 in mpi_helper.prange(start, stop, step):
-            yield p0, p1
+        yield from mpi_helper.prange(start, stop, step)
 
 
 class _ChemistsERIs(ragf2._ChemistsERIs):

diff --git a/pyscf/agf2/mpi_helper.py b/pyscf/agf2/mpi_helper.py
@@ -130,8 +130,7 @@ def nrange(start, stop=None, step=1):
     if stop is None:
         start, stop = 0, start
 
-    for i in range(start+rank, stop, step*size):
-        yield i
+    yield from range(start+rank, stop, step*size)
 
 
 def prange(start, stop, step):

diff --git a/pyscf/pbc/lib/kpts.py b/pyscf/pbc/lib/kpts.py
@@ -1173,5 +1173,4 @@ def cache_stabilizer(self):
     def loop_stabilizer(self, index):
         if self._kqrts_stab is None or self._ops_stab is None:
             self.cache_stabilizer()
-        for klcd, iop in zip(self._kqrts_stab[index], self._ops_stab[index]):
-            yield klcd, iop
+        yield from zip(self._kqrts_stab[index], self._ops_stab[index])
diff --git a/pyscf/tdscf/common_slow.py b/pyscf/tdscf/common_slow.py
@@ -605,8 +605,7 @@ def __call__(self, v):
         return r
 
     def __iter__(self):
-        for i, o, e in zip(self.args, self.results, self.errors):
-            yield i, o, e
+        yield from zip(self.args, self.results, self.errors)
 
     @property
     def ncalls(self):