Skip to content

Commit

Permalink
[python] name pypolars
Browse files Browse the repository at this point in the history
  • Loading branch information
ritchie46 committed Sep 17, 2020
1 parent 98da19b commit e39a6d1
Show file tree
Hide file tree
Showing 7 changed files with 14 additions and 7 deletions.
File renamed without changes.
2 changes: 1 addition & 1 deletion py-polars/polars/ffi.py → py-polars/pypolars/ffi.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from numpy import ctypeslib
import ctypes
from typing import Any
from .polars import (
from .pypolars import (
aligned_array_f32,
aligned_array_f64,
aligned_array_i32,
Expand Down
2 changes: 1 addition & 1 deletion py-polars/polars/frame.py → py-polars/pypolars/frame.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
from __future__ import annotations
from .polars import PyDataFrame, PySeries
from .pypolars import PyDataFrame, PySeries
from typing import Dict, Sequence, List, Tuple, Optional, Union
from .series import Series, wrap_s
import numpy as np
Expand Down
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
from __future__ import annotations
from .polars import PySeries
from .pypolars import PySeries
import numpy as np
from typing import Optional, List, Sequence, Union, Any
from .ffi import ptr_to_numpy, aligned_array_f64, series_from_ptr_f64
Expand Down
9 changes: 8 additions & 1 deletion py-polars/src/lib.rs
Original file line number Diff line number Diff line change
Expand Up @@ -21,10 +21,17 @@ macro_rules! create_aligned_buffer {
};
}

create_aligned_buffer!(aligned_array_f16, f16);
create_aligned_buffer!(aligned_array_f32, f32);
create_aligned_buffer!(aligned_array_f64, f64);
create_aligned_buffer!(aligned_array_i8, i8);
create_aligned_buffer!(aligned_array_i16, i16);
create_aligned_buffer!(aligned_array_i32, i32);
create_aligned_buffer!(aligned_array_i64, i64);
create_aligned_buffer!(aligned_array_u8, u8);
create_aligned_buffer!(aligned_array_u16, u16);
create_aligned_buffer!(aligned_array_u32, u32);
create_aligned_buffer!(aligned_array_u64, u64);

#[pyfunction]
pub fn series_from_ptr_f64(name: &str, ptr: usize, len: usize) -> PySeries {
Expand All @@ -35,7 +42,7 @@ pub fn series_from_ptr_f64(name: &str, ptr: usize, len: usize) -> PySeries {
}

#[pymodule]
fn polars(_py: Python, m: &PyModule) -> PyResult<()> {
fn pypolars(_py: Python, m: &PyModule) -> PyResult<()> {
m.add_class::<PySeries>().unwrap();
m.add_class::<PyDataFrame>().unwrap();
m.add_wrapped(wrap_pyfunction!(aligned_array_f32)).unwrap();
Expand Down
2 changes: 1 addition & 1 deletion py-polars/tests/test_df.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
from polars import DataFrame, Series
from pypolars import DataFrame, Series
import pytest


Expand Down
4 changes: 2 additions & 2 deletions py-polars/tests/test_series.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
from polars import Series
from polars.ffi import aligned_array_f32
from pypolars import Series
from pypolars.ffi import aligned_array_f32
import numpy as np


Expand Down

0 comments on commit e39a6d1

Please sign in to comment.