Add comments and code cleanup

vccimaging · Apr 10, 2023 · 5ed56fa · 5ed56fa
1 parent 2dd118b
commit 5ed56fa
Show file tree

Hide file tree

Showing 4 changed files with 286 additions and 151 deletions.
diff --git a/diffoptics/basics.py b/diffoptics/basics.py
@@ -1,4 +1,3 @@
-import math
 from enum import Enum
 import torch
 import numpy as np
@@ -186,9 +185,9 @@ def __init__(self, name=None):
 
  # This table is hard-coded. TODO: Import glass libraries from Zemax.
  self.MATERIAL_TABLE = { # [nD, Abbe number]
- "vacuum": [1., math.inf],
- "air": [1.000293, math.inf],
- "occluder": [1., math.inf],
+ "vacuum": [1., np.inf],
+ "air": [1.000293, np.inf],
+ "occluder": [1., np.inf],
  "f2": [1.620, 36.37],
  "f15": [1.60570, 37.831],
  "uvfs": [1.458, 67.82],