Skip to content

Commit

Permalink
chore: fix pcc
Browse files Browse the repository at this point in the history
  • Loading branch information
jfrery committed May 23, 2024
1 parent 16e5e90 commit d7ef157
Show file tree
Hide file tree
Showing 4 changed files with 8 additions and 3 deletions.
1 change: 1 addition & 0 deletions .gitleaksignore
Original file line number Diff line number Diff line change
Expand Up @@ -8,3 +8,4 @@ a99389ee01cbb972e46a892d3d0e9c7f8ee23f59:use_case_examples/training/analyze.ipyn
a99389ee01cbb972e46a892d3d0e9c7f8ee23f59:use_case_examples/training/analyze.ipynb:aws-access-token:18379
f41de03048a9ed27946b875e81b34138bb4bb17b:use_case_examples/training/analyze.ipynb:aws-access-token:6404
e2904473898ddd325f245f4faca526a0e9520f49:builders/Dockerfile.zamalang-env:generic-api-key:5
7d5e885816f1f1e432dd94da38c5c8267292056a:docs/advanced_examples/XGBRegressor.ipynb:aws-access-token:1026
2 changes: 1 addition & 1 deletion Makefile
Original file line number Diff line number Diff line change
Expand Up @@ -532,7 +532,7 @@ pytest_codeblocks_one:
.PHONY: shell_lint # Lint all bash scripts
shell_lint:
@# grep -v "^\./\." is to avoid files in .hidden_directories
find . -type f -name "*.sh" | grep -v "^\./\." | \
find . -type f -name "*.sh" | grep -v "^\./\." | grep -v "^\./playground/" | \
xargs shellcheck

.PHONY: set_version # Generate a new version number and update all files with it accordingly
Expand Down
6 changes: 5 additions & 1 deletion src/concrete/ml/quantization/quantized_ops.py
Original file line number Diff line number Diff line change
Expand Up @@ -1586,7 +1586,10 @@ class QuantizedDiv(QuantizedMixingOp):
_impl_for_op_named: str = "Div"

def __init__(
self, *args, rounding_threshold_bits: int | Dict[str, str | int] | None = None, **kwargs
self,
*args,
rounding_threshold_bits: Union[None, int, Dict[str, Union[str, int]]] = None,
**kwargs,
) -> None:
super().__init__(*args, rounding_threshold_bits=rounding_threshold_bits, **kwargs)
self.divider_quantizer: UniformQuantizer
Expand Down Expand Up @@ -1711,6 +1714,7 @@ def q_impl(
params=self.output_quant_params,
)


class QuantizedMul(QuantizedMixingOp):
"""Quantized Multiplication operator.
Expand Down
2 changes: 1 addition & 1 deletion tests/torch/test_compile_torch.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,8 +48,8 @@
StepActivationModule,
TinyQATCNN,
TorchDivide,
UnivariateModule,
TorchMultiply,
UnivariateModule,
)
from concrete.ml.quantization import QuantizedModule

Expand Down

0 comments on commit d7ef157

Please sign in to comment.