update get_ngk func

haozhihan · haozhihan · commit 588a335da019 · 2024-12-29T13:01:13.000Z
diff --git a/source/module_psi/psi.cpp b/source/module_psi/psi.cpp
@@ -484,10 +484,7 @@ int Psi<T, Device>::get_current_nbas() const
 template <typename T, typename Device>
 const int& Psi<T, Device>::get_ngk(const int ik_in) const
 {
-    if (!this->ngk)
-    {
-        return this->nbasis;
-    }
+    assert(this->ngk != nullptr);
     return this->ngk[ik_in];
 }
 

Original file line number	Diff line number	Diff line change
`@@ -484,10 +484,7 @@ int Psi<T, Device>::get_current_nbas() const`
`484`	`484`	`template <typename T, typename Device>`
`485`	`485`	`const int& Psi<T, Device>::get_ngk(const int ik_in) const`
`486`	`486`	`{`
`487`		`- if (!this->ngk)`
`488`		`- {`
`489`		`- return this->nbasis;`
`490`		`- }`
	`487`	`+ assert(this->ngk != nullptr);`
`491`	`488`	`return this->ngk[ik_in];`
`492`	`489`	`}`
`493`	`490`