Skip to content

Commit

Permalink
removing final numba tags
Browse files Browse the repository at this point in the history
  • Loading branch information
maxdolan authored and maxdolan committed Jun 27, 2024
1 parent 98bc324 commit e83bd8d
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions pylj/pairwise.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,5 @@
from __future__ import division
import numpy as np
from numba import njit, jit
from pylj import pairwise as heavy

#Jit tag here had to be removed
Expand Down Expand Up @@ -56,7 +55,7 @@ def compute_force(particles, box_length, cut_off, constants, forcefield, mass):
particles = update_accelerations(particles, forces, mass_kg, dx, dy, distances)
return particles, distances, forces, energies

@njit
#Jit tag here had to be removed
def separation(dx, dy):
"""Calculate the distance in 2D space.
Parameters
Expand Down Expand Up @@ -290,7 +289,7 @@ def heat_bath(particles, temperature_sample, bath_temp):
return particles


@njit
#Jit tag here had to be removed
def dist(xposition, yposition, box_length):
"""Returns the distance array for the set of particles.
Parameters
Expand Down Expand Up @@ -330,7 +329,7 @@ def dist(xposition, yposition, box_length):
return drr, dxr, dyr


@njit
#Jit tag here had to be removed
def pbc_correction(position, cell):
"""Correct for the periodic boundary condition.
Parameters
Expand Down

0 comments on commit e83bd8d

Please sign in to comment.