fix nrSymbolDemodulate

catkira · Oct 29, 2023 · 3292a67 · 3292a67
1 parent fdad89e
commit 3292a67
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/py3gpp/nrSymbolDemodulate.py b/py3gpp/nrSymbolDemodulate.py
@@ -39,15 +39,15 @@ def nrSymbolDemodulate(input, mod, nVar=1e-10, DecisionType="soft"):
         elif mod == "QPSK":
             output = np.append(output, np.real(symbol))
             output = np.append(output, np.imag(symbol))
-        elif mod == "QAM16":
+        elif mod == "16QAM":
             output = np.append(output, np.real(symbol))
             output = np.append(output, np.imag(symbol))
             output = np.append(output, -(np.abs(np.real(symbol)) - 2 / np.sqrt(10)))
             output = np.append(output, -(np.abs(np.imag(symbol)) - 2 / np.sqrt(10)))
 
     if DecisionType == "soft":
         output /= nVar / np.exp(1)
-        if mod == "QAM16":
+        if mod == "16QAM":
             output /= 2
     else:
         output = (output > 0).astype(int)