diff --git a/priorConditionedAnnealing/noise.py b/priorConditionedAnnealing/noise.py
index 2e9d0de..a87cfd7 100644
--- a/priorConditionedAnnealing/noise.py
+++ b/priorConditionedAnnealing/noise.py
@@ -114,8 +114,11 @@ class Harmonic_Perlin_Noise():
     def __init__(self, known_shape=None, scale=0.1, octaves=8):
         self.known_shape = known_shape
         self.scale = scale
-        if type(octaves) == int:
-            octaves = [1/(i+1) for i in range(octaves)]
+        if type(octaves) in [int, float]:
+            int_octaves = int(octaves)
+            octaves = [1/(i+1) for i in range(int_octaves)]
+            if type(octaves) == float:
+                octaves += [1/(int_octaves+2)*(octaves-int_octaves)]
         octaves = np.array(octaves)
         self.octaves = octaves / np.linalg.norm(octaves)
         self.reset()